Update app.py

app.py

@@ -1,8 +1,8 @@
 import gradio as gr
 
-import torch
+from ultralytics import YOLO
 
-model = torch.hub.load("ultralytics/yolov5", "best")
+model = YOLO("best.pt")
 
 def greet(source):
   model = YOLO("best.pt")