innat commited on
Commit
104d3e6
·
1 Parent(s): 6f09df4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -55,10 +55,10 @@ def get_model(model_type):
55
  print('innat/videomae/' + model_type + '_FT')
56
 
57
  ft_path = hf_hub_download(
58
- repo_id=hf_dataset_identifier, filename='innat/videomae/' + model_type + '_FT', repo_type="model"
59
  )
60
  pt_path = hf_hub_download(
61
- repo_id=hf_dataset_identifier, filename='innat/videomae/' + model_type + '_PT', repo_type="model"
62
  )
63
 
64
  ft_model = keras.models.load_model(ft_path)
 
55
  print('innat/videomae/' + model_type + '_FT')
56
 
57
  ft_path = hf_hub_download(
58
+ repo_id='innat/videomae', filename=model_type + '_FT', repo_type="model"
59
  )
60
  pt_path = hf_hub_download(
61
+ repo_id='innat/videomae', filename=model_type + '_PT', repo_type="model"
62
  )
63
 
64
  ft_model = keras.models.load_model(ft_path)