Spaces:
Runtime error
Runtime error
Move numpy import
Browse files
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")
|