Updated requirements

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import spaces
 import PIL.Image as Image
-from ultralytics import ASSETS, YOLO
+from ultralytics import YOLO
 
 
 
@@ -13,7 +13,6 @@ def predict_image(img, conf_threshold, iou_threshold):
     
     results = model.predict(
         source=img,
-        device='cuda:0',
         conf=conf_threshold,
         iou=iou_threshold,
         show_labels=True,

requirements.txt

@@ -1,4 +1,3 @@
 ultralytics
-torch
-supervision
-spaces
+spaces
+gradio