Upload 5 files

app.py

@@ -416,9 +416,20 @@ def start_training(epochs, batch_size, learning_rate):
     ZeroGPU: starting it from a normal Python background thread can bypass the
     GPU allocation context, making the button appear to do nothing.
     """
+    global _training
     if _training["running"]:
         return json.dumps(_training, indent=2)
-    # Basic validation before requesting GPU time.
+    # Validate parameters before requesting GPU time.
+    try:
+        epochs = max(1, int(epochs))
+        batch_size = max(1, int(batch_size))
+        learning_rate = float(learning_rate)
+        if learning_rate <= 0:
+            raise ValueError("Learning rate must be greater than 0.")
+    except Exception as e:
+        _training = {"running": False, "message": "training failed", "error": f"Invalid training settings: {e}"}
+        return json.dumps(_training, indent=2)
+
     annotated = sum(bool(x.get("annotations")) for x in load_dataset()["images"])
     if annotated < 1:
         _training = {"running": False, "message": "training failed", "error": "Annotate at least 1 image before training."}