Hugging Face's logo

Spaces:
kedimestan
/
od-test-1
Runtime error

App Files Community
kedimestan commited on
Commit
be4cefb
·
verified ·
1 Parent(s): 92ce21f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import io
7
  app = FastAPI()
8
 
9
  # Load the YOLOv5 model for object detection
10
- detector = pipeline('object-detection', model='ultralytics/yolov5')
11
 
12
  @app.post("/detect_objects/")
13
  async def detect_objects(file: UploadFile = File(...)):
 
7
  app = FastAPI()
8
 
9
  # Load the YOLOv5 model for object detection
10
+ detector = pipeline('object-detection', model='Ultralytics/YOLOv8')
11
 
12
  @app.post("/detect_objects/")
13
  async def detect_objects(file: UploadFile = File(...)):