krunakuamar commited on
Commit
ae07b63
1 Parent(s): 4eb40e9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -275,8 +275,8 @@ def init_data():
275
  model_device = "cpu"
276
  global yolo_model
277
  # TODO Update for local development
278
- yolo_model = YOLO('yolov8x-seg.pt')
279
- # yolo_model = YOLO('/app/yolov8x-seg.pt')
280
  yolo_model.to(model_device)
281
  print(f"YOLO model yolov8x-seg.pt loaded.")
282
  server.initModel()
 
275
  model_device = "cpu"
276
  global yolo_model
277
  # TODO Update for local development
278
+ # yolo_model = YOLO('yolov8x-seg.pt')
279
+ yolo_model = YOLO('/app/yolov8x-seg.pt')
280
  yolo_model.to(model_device)
281
  print(f"YOLO model yolov8x-seg.pt loaded.")
282
  server.initModel()