Spaces:
Runtime error
Runtime error
krunakuamar
commited on
Commit
•
ae07b63
1
Parent(s):
4eb40e9
Update app.py
Browse files
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 |
-
|
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()
|