k23064919 commited on
Commit
3edadc3
·
1 Parent(s): 65f179b

update modelLoader

Browse files
Files changed (2) hide show
  1. ui/config.py +1 -1
  2. 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": "12922d47f3be4d14856a84cdabfe5cb5"
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 or 'clearml_task_id' not in modelConfig:
18
  raise ValueError(f"ClearML configuration not found for model: {modelName}")
19
 
20
- taskID = modelConfig['clearmml_task_id']
21
- modelType = modelConfig['modelType']
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}")