Migrate to ZeroGPU and update requirements for compatibility

app.py

@@ -4,6 +4,7 @@ from fastapi import FastAPI
 from PIL import Image
 import torch
 import spaces
+import numpy as np
 
 # 初始化 FastAPI 和模型
 app = FastAPI()
@@ -14,8 +15,9 @@ model = YOLO('NailongKiller.yolo11n.pt').to(device)
 
 @spaces.GPU
 def predict(img):
-    # 使用 PIL 直接转换为 PyTorch 张量
-    img_resized = Image.fromarray(img).resize((640, 640))
+    # 将 PIL 图像转换为 numpy 数组
+    img_resized = np.array(Image.fromarray(img).resize((640, 640)))
+    # 将 numpy 数组转换为 PyTorch 张量
     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()