Spaces:
Sleeping
Sleeping
Update main.py
Browse files
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()
|