Update app.py

app.py

@@ -63,7 +63,7 @@ def auto_train():
 
         # Determine the model name based on the selected size and task
         model_type = "seg" if task == "Segmentation" else "cls"
-        model_name = f"yolov8{model_size[0]}-{model_type}.pt"
+        model_name = f"yolov8{model_size}-{model_type}.pt"
 
         # Load and train the model
         model = YOLO(model_name)