osanseviero commited on
Commit
68e0199
·
1 Parent(s): d5e9a49

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -20,8 +20,9 @@ def make_clickable_user(user_id):
20
 
21
  def get_model_ids(assignment):
22
  api = HfApi()
23
- models = api.list_models(author="Classroom-workshop", filter=assignment)
24
  model_ids = [x.modelId for x in models]
 
25
  return model_ids
26
 
27
  def get_metadata(model_id):
 
20
 
21
  def get_model_ids(assignment):
22
  api = HfApi()
23
+ models = api.list_models(author="Classroom-workshop", search=assignment)
24
  model_ids = [x.modelId for x in models]
25
+ print(model_ids)
26
  return model_ids
27
 
28
  def get_metadata(model_id):