app.py

app.py

@@ -25,8 +25,8 @@ def train_model(dataset_file: gr.File, batch_size: int, epochs: int, lr: float,
     os.makedirs(MODEL_OUTPUT_DIR, exist_ok=True)
 
     # 2. File Handling: Use the temporary path of the uploaded file
-    if dataset_file is None or not dataset_file.path.endswith(".json"):
-        return "❌ ERROR: Please upload a valid Label Studio JSON file.", None
+    # if dataset_file is None or not dataset_file.path.endswith(".json"):
+    #     return "❌ ERROR: Please upload a valid Label Studio JSON file.", None
 
     input_path = dataset_file.path