Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,6 @@ from utils import IMAGENET_MEAN, IMAGENET_STD, num_frames, patch_size, input_siz
|
|
11 |
from labels import K400_label_map, SSv2_label_map, UCF_label_map
|
12 |
|
13 |
|
14 |
-
MODEL_CACHE = {}
|
15 |
MODELS = {
|
16 |
'K400': [
|
17 |
'innat/videomae/TFVideoMAE_S_K400_16x224_FT',
|
@@ -47,9 +46,6 @@ def video_to_gif(video_array, gif_filename):
|
|
47 |
|
48 |
def get_model(data_type):
|
49 |
|
50 |
-
if data_type in MODEL_CACHE:
|
51 |
-
return MODEL_CACHE[data_type]
|
52 |
-
|
53 |
if data_type=='K400':
|
54 |
ft_model = keras.models.load_model(MODELS[data_type][0])
|
55 |
pt_model = keras.models.load_model(MODELS[data_type][1])
|
@@ -65,7 +61,6 @@ def get_model(data_type):
|
|
65 |
pt_model = keras.models.load_model(MODELS[data_type][1])
|
66 |
label_map = {v: k for k, v in UCF_label_map.items()}
|
67 |
|
68 |
-
MODEL_CACHE[data_type] = (ft_model, pt_model, label_map)
|
69 |
|
70 |
return ft_model, pt_model, label_map
|
71 |
|
@@ -111,7 +106,7 @@ gr.Interface(
|
|
111 |
gr.Video(type="file"),
|
112 |
gr.Radio(
|
113 |
['K400', 'SSv2', 'UCF'],
|
114 |
-
label='Dataset'
|
115 |
),
|
116 |
],
|
117 |
outputs=[
|
|
|
11 |
from labels import K400_label_map, SSv2_label_map, UCF_label_map
|
12 |
|
13 |
|
|
|
14 |
MODELS = {
|
15 |
'K400': [
|
16 |
'innat/videomae/TFVideoMAE_S_K400_16x224_FT',
|
|
|
46 |
|
47 |
def get_model(data_type):
|
48 |
|
|
|
|
|
|
|
49 |
if data_type=='K400':
|
50 |
ft_model = keras.models.load_model(MODELS[data_type][0])
|
51 |
pt_model = keras.models.load_model(MODELS[data_type][1])
|
|
|
61 |
pt_model = keras.models.load_model(MODELS[data_type][1])
|
62 |
label_map = {v: k for k, v in UCF_label_map.items()}
|
63 |
|
|
|
64 |
|
65 |
return ft_model, pt_model, label_map
|
66 |
|
|
|
106 |
gr.Video(type="file"),
|
107 |
gr.Radio(
|
108 |
['K400', 'SSv2', 'UCF'],
|
109 |
+
label='Dataset', value='K400'
|
110 |
),
|
111 |
],
|
112 |
outputs=[
|