Spaces:
Sleeping
Sleeping
test
Browse files- __pycache__/main.cpython-310.pyc +0 -0
- main.py +6 -0
__pycache__/main.cpython-310.pyc
CHANGED
|
Binary files a/__pycache__/main.cpython-310.pyc and b/__pycache__/main.cpython-310.pyc differ
|
|
|
main.py
CHANGED
|
@@ -509,6 +509,12 @@ async def start_training(req: TrainRequest, background_tasks: BackgroundTasks):
|
|
| 509 |
status_code=400,
|
| 510 |
detail="Dataset backend tidak ditemukan atau sesi upload sudah expired. Silakan upload ZIP dataset lagi dari Dataset Manager.",
|
| 511 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 512 |
elif not req.images:
|
| 513 |
raise HTTPException(
|
| 514 |
status_code=400,
|
|
|
|
| 509 |
status_code=400,
|
| 510 |
detail="Dataset backend tidak ditemukan atau sesi upload sudah expired. Silakan upload ZIP dataset lagi dari Dataset Manager.",
|
| 511 |
)
|
| 512 |
+
total_images, class_distribution = _collect_dataset_summary(source_path)
|
| 513 |
+
if total_images < 2 or len(class_distribution.keys()) < 2:
|
| 514 |
+
raise HTTPException(
|
| 515 |
+
status_code=400,
|
| 516 |
+
detail="Dataset backend tidak valid untuk training (minimal 2 gambar dan 2 class). Upload ZIP dataset yang benar.",
|
| 517 |
+
)
|
| 518 |
elif not req.images:
|
| 519 |
raise HTTPException(
|
| 520 |
status_code=400,
|