First Deployment

app.py

@@ -1,16 +1,18 @@
+import gradio as gr
 import spaces
 import PIL.Image as Image
-import gradio as gr
 from ultralytics import ASSETS, YOLO
 
 
-model = YOLO("best.pt")
+
 
 @spaces.GPU
 def predict_image(img, conf_threshold, iou_threshold):
+    
+    model = YOLO("best.pt")
+    
     results = model.predict(
         source=img,
-        device="cuda:0",
         conf=conf_threshold,
         iou=iou_threshold,
         show_labels=True,

requirements.txt

@@ -1,5 +1 @@
 ultralytics
-pytorch
-torch
-torchvision
-opencv-python