Update app.py
Browse files
app.py
CHANGED
@@ -54,6 +54,8 @@ def get_model(data_type):
|
|
54 |
print()
|
55 |
print('-------------------- ', data_type)
|
56 |
print()
|
|
|
|
|
57 |
ft_model = keras.models.load_model(MODELS[data_type][0])
|
58 |
pt_model = keras.models.load_model(MODELS[data_type][1])
|
59 |
label_map = {v: k for k, v in K400_label_map.items()}
|
|
|
54 |
print()
|
55 |
print('-------------------- ', data_type)
|
56 |
print()
|
57 |
+
|
58 |
+
data_type ='K400'
|
59 |
ft_model = keras.models.load_model(MODELS[data_type][0])
|
60 |
pt_model = keras.models.load_model(MODELS[data_type][1])
|
61 |
label_map = {v: k for k, v in K400_label_map.items()}
|