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()}
|