app.py

import cv2
import torch
import numpy as np
from ultralytics import YOLO
from fastapi import FastAPI

app = FastAPI()
model = YOLO('model.pt')
#device = torch.device('cuda')
#model.to(device)

@app.get("/detect")
def test():
    return {'Hello':'world'}