Spaces:
Sleeping
Sleeping
update modelLoader
Browse files- ui/config.py +1 -1
- ui/model_loader.py +3 -3
ui/config.py
CHANGED
|
@@ -5,7 +5,7 @@ MODEL_CONFIGS = {
|
|
| 5 |
"CNN from Scratch": {
|
| 6 |
"description": "Custom CNN model trained from scratch",
|
| 7 |
"model_type": "cnn",
|
| 8 |
-
"clearml_task_id": "
|
| 9 |
},
|
| 10 |
"Transfer Learning (ResNet18)": {
|
| 11 |
"description": "Fine-tuned ResNet18 model",
|
|
|
|
| 5 |
"CNN from Scratch": {
|
| 6 |
"description": "Custom CNN model trained from scratch",
|
| 7 |
"model_type": "cnn",
|
| 8 |
+
"clearml_task_id": "fe14662da63d45bf9208fdf9856d2fcc"
|
| 9 |
},
|
| 10 |
"Transfer Learning (ResNet18)": {
|
| 11 |
"description": "Fine-tuned ResNet18 model",
|
ui/model_loader.py
CHANGED
|
@@ -14,11 +14,11 @@ class ModelLoader:
|
|
| 14 |
def loadFromClearml(self, modelName):
|
| 15 |
modelConfig = config.MODEL_CONFIGS.get(modelName)
|
| 16 |
|
| 17 |
-
if not modelConfig
|
| 18 |
raise ValueError(f"ClearML configuration not found for model: {modelName}")
|
| 19 |
|
| 20 |
-
taskID = modelConfig['
|
| 21 |
-
modelType = modelConfig['
|
| 22 |
|
| 23 |
try:
|
| 24 |
print(f"attemtping to fetch '{modelName}' from clearML task: {taskID}")
|
|
|
|
| 14 |
def loadFromClearml(self, modelName):
|
| 15 |
modelConfig = config.MODEL_CONFIGS.get(modelName)
|
| 16 |
|
| 17 |
+
if not modelConfig:
|
| 18 |
raise ValueError(f"ClearML configuration not found for model: {modelName}")
|
| 19 |
|
| 20 |
+
taskID = modelConfig['clearml_task_id']
|
| 21 |
+
modelType = modelConfig['model_type']
|
| 22 |
|
| 23 |
try:
|
| 24 |
print(f"attemtping to fetch '{modelName}' from clearML task: {taskID}")
|