Spaces:
Sleeping
Sleeping
add debug
Browse files- ui/model_loader.py +2 -1
ui/model_loader.py
CHANGED
|
@@ -26,7 +26,8 @@ class ModelLoader:
|
|
| 26 |
print(f"Attempting to fetch '{modelName}' from ClearML task: {taskID}")
|
| 27 |
|
| 28 |
task = Task.get_task(task_id=taskID)
|
| 29 |
-
|
|
|
|
| 30 |
# Fetch the artifact 'model_one.pt'
|
| 31 |
artifact = task.artifacts.get(MODEL_ARTIFACT_NAME)
|
| 32 |
|
|
|
|
| 26 |
print(f"Attempting to fetch '{modelName}' from ClearML task: {taskID}")
|
| 27 |
|
| 28 |
task = Task.get_task(task_id=taskID)
|
| 29 |
+
print("Available artifacts:", task.artifacts.keys())
|
| 30 |
+
|
| 31 |
# Fetch the artifact 'model_one.pt'
|
| 32 |
artifact = task.artifacts.get(MODEL_ARTIFACT_NAME)
|
| 33 |
|