Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
|
@@ -8,7 +8,7 @@ from ultralytics import YOLO
|
|
| 8 |
from datetime import datetime
|
| 9 |
|
| 10 |
app = FastAPI(title="PCB Defect Detection API")
|
| 11 |
-
model = YOLO("
|
| 12 |
|
| 13 |
class ImageRequest(BaseModel):
|
| 14 |
"""Request model for image processing endpoint."""
|
|
|
|
| 8 |
from datetime import datetime
|
| 9 |
|
| 10 |
app = FastAPI(title="PCB Defect Detection API")
|
| 11 |
+
model = YOLO("Best_defects.pt")
|
| 12 |
|
| 13 |
class ImageRequest(BaseModel):
|
| 14 |
"""Request model for image processing endpoint."""
|