Migrate to ZeroGPU and update requirements for compatibility

app.py

@@ -4,12 +4,19 @@ from fastapi import FastAPI, File, UploadFile
 from PIL import Image
 import numpy as np
 import io
+import torch
+import spaces  # 导入 spaces 模块
 
 # 初始化 FastAPI 和模型
 app = FastAPI()
-model = YOLO('NailongKiller.yolo11n.pt')
 
+# 检查 GPU 是否可用，并选择设备
+device = 'cuda' if torch.cuda.is_available() else 'cpu'
+model = YOLO('NailongKiller.yolo11n.pt').to(device)
+
+@spaces.GPU  # 使用装饰器标记需要 GPU 的函数
 def predict(img):
+    img = img.to(device)
     results = model.predict(img)
     return results[0].plot()
 
@@ -33,16 +40,15 @@ demo = gr.Interface(
 # API 端点
 @app.post("/detect/")
 async def detect_api(file: UploadFile = File(...)):
-    # 读取上传的图片
     contents = await file.read()
     image = Image.open(io.BytesIO(contents))
     image_np = np.array(image)
     
-    # 运行推理
+    image_np = torch.from_numpy(image_np).to(device)
+    
     results = model.predict(image_np)
     result = results[0]
     
-    # 返回检测结果
     detections = []
     for box in result.boxes:
         detection = {