Update app.py

app.py

@@ -92,7 +92,7 @@ def trigger_incremental_train():
     samples = list(collection.find())
     samples = [s for s in samples if "image_path" in s and "true_label" in s]
 
-    if len(samples) < 100:
+    if len(samples) < 1:
         return "Not enough validated samples (need at least 100)."
 
     m1 = UNet().to(device)
@@ -141,7 +141,8 @@ def trigger_incremental_train():
 
     return {
         "Old Model": eval_old,
-        "New Model": eval_new
+        "New Model": eval_new,
+        "status": "Training completed successfully"
     }
 
 # Gradio interface