abdrabo01 commited on
Commit
509deed
·
verified ·
1 Parent(s): bb14d85

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
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("yolo_modeln11_1502.pt")
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."""