Update main.py

main.py

@@ -1,4 +1,4 @@
-from ultralyticsplus import YOLO
+from ultralytics import YOLO
 from base64 import b64encode
 from speech_recognition import AudioFile, Recognizer
 import numpy as np
@@ -9,8 +9,8 @@ from fastapi import FastAPI, File, UploadFile, Form
 from utils import tts, read_image_file, pil_to_base64, base64_to_pil, get_hist
 from typing import Optional
 
-# model = YOLO('ultralyticsplus/yolov8s')
-# CLASS = model.model.names
+model = YOLO('code/yolov8s.pt')
+CLASS = model.model.names
 
 app = FastAPI()
 defaul_bot_voice = "おはいようございます"