Migrate to ZeroGPU and update requirements for compatibility

app.py

@@ -16,8 +16,9 @@ model = YOLO('NailongKiller.yolo11n.pt').to(device)
 
 @spaces.GPU  # 使用装饰器标记需要 GPU 的函数
 def predict(img):
-    img = img.to(device)
-    results = model.predict(img)
+    # 将 numpy 数组转换为 PyTorch 张量
+    img_tensor = torch.from_numpy(img).to(device)
+    results = model.predict(img_tensor)
     return results[0].plot()
 
 # Gradio 界面
@@ -44,9 +45,10 @@ async def detect_api(file: UploadFile = File(...)):
     image = Image.open(io.BytesIO(contents))
     image_np = np.array(image)
     
-    image_np = torch.from_numpy(image_np).to(device)
+    # 将图像移动到 GPU
+    image_tensor = torch.from_numpy(image_np).to(device)
     
-    results = model.predict(image_np)
+    results = model.predict(image_tensor)
     result = results[0]
     
     detections = []