Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Commit
·
0b9adca
1
Parent(s):
f6f28fc
Update model submission checks: include 'task' in pending and finished model requests
Browse files
utils.py
CHANGED
|
@@ -75,7 +75,7 @@ def submit_model(model_name, revision, precision, params, license, task):
|
|
| 75 |
|
| 76 |
# Check if model is in pending requests
|
| 77 |
if not df_pending.empty:
|
| 78 |
-
existing_models_pending = df_pending[['model_name', 'revision', 'precision']]
|
| 79 |
model_exists_in_pending = ((existing_models_pending['model_name'] == model_name) &
|
| 80 |
(existing_models_pending['revision'] == revision) &
|
| 81 |
(existing_models_pending['precision'] == precision) &
|
|
@@ -85,7 +85,7 @@ def submit_model(model_name, revision, precision, params, license, task):
|
|
| 85 |
|
| 86 |
# Check if model is in finished requests
|
| 87 |
if not df_finished.empty:
|
| 88 |
-
existing_models_finished = df_finished[['model_name', 'revision', 'precision']]
|
| 89 |
model_exists_in_finished = ((existing_models_finished['model_name'] == model_name) &
|
| 90 |
(existing_models_finished['revision'] == revision) &
|
| 91 |
(existing_models_finished['precision'] == precision) &
|
|
|
|
| 75 |
|
| 76 |
# Check if model is in pending requests
|
| 77 |
if not df_pending.empty:
|
| 78 |
+
existing_models_pending = df_pending[['model_name', 'revision', 'precision', 'task']]
|
| 79 |
model_exists_in_pending = ((existing_models_pending['model_name'] == model_name) &
|
| 80 |
(existing_models_pending['revision'] == revision) &
|
| 81 |
(existing_models_pending['precision'] == precision) &
|
|
|
|
| 85 |
|
| 86 |
# Check if model is in finished requests
|
| 87 |
if not df_finished.empty:
|
| 88 |
+
existing_models_finished = df_finished[['model_name', 'revision', 'precision', 'task']]
|
| 89 |
model_exists_in_finished = ((existing_models_finished['model_name'] == model_name) &
|
| 90 |
(existing_models_finished['revision'] == revision) &
|
| 91 |
(existing_models_finished['precision'] == precision) &
|