quickfix

ui/model_loader.py

@@ -7,7 +7,7 @@ from models.modelOne import modelOne
 
 sys.path.append(str(Path(__file__).parent.parent))
 
-MODEL_ARTIFACT_NAME = 'model_one.pt'
+MODEL_ARTIFACT_NAME = 'best_model'
 
 class ModelLoader:
     def __init__(self):
@@ -27,7 +27,7 @@ class ModelLoader:
 
             task = Task.get_task(task_id=taskID)
             print("Available artifacts:", task.artifacts.keys())
-            
+
             # Fetch the artifact 'model_one.pt'
             artifact = task.artifacts.get(MODEL_ARTIFACT_NAME)