Hugging Face's logo

Spaces:
innat
/
VideoMAE
Runtime error

App Files Community
innat commited on
Commit
d1b91e1
·
1 Parent(s): 36e374f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -61,8 +61,10 @@ def tube_mask_generator(mask_ratio):
61
 
62
 
63
  def get_model(model_type):
64
- ft_model = keras.models.load_model(model_type + '_FT')
65
- pt_model = keras.models.load_model(model_type + '_PT')
 
 
66
 
67
  if 'K400' in model_type:
68
  data_type = 'K400'
 
61
 
62
 
63
  def get_model(model_type):
64
+ print('---------------------')
65
+ print('innat/videomae/' + model_type + '_FT')
66
+ ft_model = keras.models.load_model('innat/videomae/' + model_type + '_FT')
67
+ pt_model = keras.models.load_model('innat/videomae/' + model_type + '_PT')
68
 
69
  if 'K400' in model_type:
70
  data_type = 'K400'