fcakyon commited on
Commit
d265161
1 Parent(s): 1651019

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -6,7 +6,7 @@ from utils import (
6
  get_num_total_frames,
7
  )
8
  from transformers import pipeline
9
- from huggingface_hub import HfApi, ModelSearchArguments, ModelFilter
10
 
11
  FRAME_SAMPLING_RATE = 4
12
  DEFAULT_MODEL = "facebook/timesformer-base-finetuned-k400"
@@ -60,10 +60,9 @@ examples = [
60
 
61
 
62
  def get_video_model_names():
63
- model_args = ModelSearchArguments()
64
  filter = ModelFilter(
65
- task=model_args.pipeline_tag.VideoClassification,
66
- library=model_args.library.Transformers,
67
  )
68
  api = HfApi()
69
  video_models = list(
 
6
  get_num_total_frames,
7
  )
8
  from transformers import pipeline
9
+ from huggingface_hub import HfApi, ModelFilter
10
 
11
  FRAME_SAMPLING_RATE = 4
12
  DEFAULT_MODEL = "facebook/timesformer-base-finetuned-k400"
 
60
 
61
 
62
  def get_video_model_names():
 
63
  filter = ModelFilter(
64
+ task='video-classification',
65
+ library='transformers',
66
  )
67
  api = HfApi()
68
  video_models = list(