Update main.py

main.py

@@ -8,13 +8,14 @@ from sahi.utils.cv import read_image_as_pil
 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
+from huggingface_hub import hf_hub_download
 
-# model = YOLO('/home/user/app/yolov8s.pt')
-# CLASS = model.model.names
+model_path = hf_hub_download(repo_id="ultralyticsplus/yolov8s", filename='yolov8s.pt')
+model = YOLO(model_path)
+CLASS = model.model.names
+defaul_bot_voice = "おはいようございます"
+area_thres = 0.3
 
-# app = FastAPI()
-# defaul_bot_voice = "おはいようございます"
-# area_thres = 0.3
 app = FastAPI()
 
 @app.get("/")