Update ui_routes.py

huggingface · May 17, 2024 · 9c2c7b5 · 9c2c7b5
1 parent 2598aad
commit 9c2c7b5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/autotrain/app/ui_routes.py b/src/autotrain/app/ui_routes.py
@@ -284,7 +284,7 @@ async def handle_form(
                 percent_valid=None,  # TODO: add to UI
                 local=hardware.lower() == "local-ui",
             )
-        if task == "image-object-detection":
+        elif task == "image-object-detection":
             dset = AutoTrainObjectDetectionDataset(
                 train_data=training_files[0],
                 token=token,