omermazig commited on
Commit
895ad89
1 Parent(s): d0339fe

Move numpy import

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -1,7 +1,5 @@
1
  import gradio as gr
2
  import tempfile
3
-
4
- import numpy as np
5
  import torch
6
  from pytorchvideo.data import make_clip_sampler
7
  from pytorchvideo.data.clip_sampling import ClipInfoList
@@ -19,6 +17,7 @@ from torchvision.transforms import (
19
  from transformers import VideoMAEForVideoClassification, VideoMAEFeatureExtractor
20
 
21
  from video_utils import change_video_resolution_and_fps
 
22
 
23
  MODEL_CKPT = "omermazig/videomae-finetuned-nba-5-class-4-batch-8000-vid-multilabel"
24
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
1
  import gradio as gr
2
  import tempfile
 
 
3
  import torch
4
  from pytorchvideo.data import make_clip_sampler
5
  from pytorchvideo.data.clip_sampling import ClipInfoList
 
17
  from transformers import VideoMAEForVideoClassification, VideoMAEFeatureExtractor
18
 
19
  from video_utils import change_video_resolution_and_fps
20
+ import numpy as np
21
 
22
  MODEL_CKPT = "omermazig/videomae-finetuned-nba-5-class-4-batch-8000-vid-multilabel"
23
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")