Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -52,7 +52,7 @@ def add_new_eval(
|
|
52 |
requests= load_dataset("EnergyStarAI/requests_debug", split="test", token=TOKEN)
|
53 |
requests_dset = requests.to_pandas()
|
54 |
model_list= requests_dset[requests_dset['status'] == 'COMPLETED']['model'].tolist
|
55 |
-
task_models = list(
|
56 |
task_model_names = [m.id for m in task_models]
|
57 |
if repo_id in model_list:
|
58 |
return 'This model has already been run!'
|
|
|
52 |
requests= load_dataset("EnergyStarAI/requests_debug", split="test", token=TOKEN)
|
53 |
requests_dset = requests.to_pandas()
|
54 |
model_list= requests_dset[requests_dset['status'] == 'COMPLETED']['model'].tolist
|
55 |
+
task_models = list(API.list_models(filter=task.lower().replace(' ','-')))
|
56 |
task_model_names = [m.id for m in task_models]
|
57 |
if repo_id in model_list:
|
58 |
return 'This model has already been run!'
|