Spaces:
Runtime error
Runtime error
Commit
•
59ab830
1
Parent(s):
457d9a8
Add onnx
Browse files- background_task.py +1 -1
background_task.py
CHANGED
@@ -200,7 +200,7 @@ def get_models_list(filter_bad_models) -> list:
|
|
200 |
models = []
|
201 |
models_ids = []
|
202 |
data = pd.read_csv(os.path.join(DATASET_REPO_URL, "resolve", "main", ELO_FILENAME))
|
203 |
-
models_on_hub = api.list_models(filter=["reinforcement-learning", "ml-agents", "ML-Agents-SoccerTwos"])
|
204 |
for i, row in data.iterrows():
|
205 |
model_id = row["author"] + "/" + row["model"]
|
206 |
if model_id in filter_bad_models:
|
|
|
200 |
models = []
|
201 |
models_ids = []
|
202 |
data = pd.read_csv(os.path.join(DATASET_REPO_URL, "resolve", "main", ELO_FILENAME))
|
203 |
+
models_on_hub = api.list_models(filter=["reinforcement-learning", "ml-agents", "ML-Agents-SoccerTwos", "onnx"])
|
204 |
for i, row in data.iterrows():
|
205 |
model_id = row["author"] + "/" + row["model"]
|
206 |
if model_id in filter_bad_models:
|