Migrate to ZeroGPU and update requirements for compatibility

app.py

@@ -2,7 +2,6 @@ import gradio as gr
 from ultralytics import YOLO
 from fastapi import FastAPI
 from PIL import Image
-import numpy as np
 import torch
 import spaces
 
@@ -15,8 +14,9 @@ model = YOLO('NailongKiller.yolo11n.pt').to(device)
 
 @spaces.GPU
 def predict(img):
-    img_resized = np.array(Image.fromarray(img).resize((640, 640)))
-    img_tensor = torch.from_numpy(img_resized).permute(2, 0, 1).unsqueeze(0).to(device)
+    # 使用 PIL 直接转换为 PyTorch 张量
+    img_resized = Image.fromarray(img).resize((640, 640))
+    img_tensor = torch.tensor(img_resized, dtype=torch.float32).permute(2, 0, 1).unsqueeze(0).to(device)
     results = model.predict(img_tensor)
     return results[0].plot()
 

requirements.txt

@@ -1,8 +1,7 @@
-torch==2.0.1
-torchvision>=0.9.0
+torch
+torchvision
 ultralytics>=8.3.0
 Pillow
 fastapi
 python-multipart
-uvicorn
-numpy
+uvicorn