aisatsu-api / main.py
vumichien's picture
Update main.py
39f22bf
raw
history blame
610 Bytes
from ultralytics import YOLO
from base64 import b64encode
from speech_recognition import AudioFile, Recognizer
import numpy as np
from scipy.spatial import distance as dist
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
model = YOLO('/home/user/app/yolov8s.pt')
CLASS = model.model.names
app = FastAPI()
defaul_bot_voice = "γŠγ―γ„γ‚ˆγ†γ”γ–γ„γΎγ™"
area_thres = 0.3
app = FastAPI()
@app.get("/")
def read_root():
return {"Hello": "World!"}