innat commited on
Commit
9fca693
1 Parent(s): 74a5ace

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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()}