vumichien commited on
Commit
39f22bf
·
1 Parent(s): 409bb27

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -9,7 +9,7 @@ from fastapi import FastAPI, File, UploadFile, Form
9
  from utils import tts, read_image_file, pil_to_base64, base64_to_pil, get_hist
10
  from typing import Optional
11
 
12
- model = YOLO('yolov8s.pt')
13
  CLASS = model.model.names
14
 
15
  app = FastAPI()
 
9
  from utils import tts, read_image_file, pil_to_base64, base64_to_pil, get_hist
10
  from typing import Optional
11
 
12
+ model = YOLO('/home/user/app/yolov8s.pt')
13
  CLASS = model.model.names
14
 
15
  app = FastAPI()