quickfix

ui/model_loader.py

@@ -22,16 +22,12 @@ class ModelLoader:
         
         taskID = modelConfig['clearml_task_id']
     
-
-        task = Task.get_task(task_id=taskID)
-
-        if not task:
-            raise ValueError(f"ClearML Task ID {taskID} not found or inaccessible.")
+        modelObject = Model(task_id=taskID)
 
         try:
             print(f"attemtping to fetch '{modelName}' from clearML task: {taskID}")
 
-            modelPath = Model.get_local_copy(task_id=taskID, file_name=MODEL_ARTIFACT_NAME)
+            modelPath = Model.get_local_copy(file_name=MODEL_ARTIFACT_NAME)
             print(f"Weights downloaded to local path: {modelPath}")
 
             model = modelOne(noOfClasses=39)