Update app.py

app.py

@@ -1,16 +1,16 @@
 import gradio as gr
+from ultralytics import YOLO
 import cv2
-import torch
 
-model = torch.hub.load('ultralytics/yolov8', 'custom', path='best.pt')
+model = YOLO("last.pt")
 
-def detect(image):
-    results = model(image)
-    return results.render()[0]
+def predict(image):
+    results = model(image)[0]
+    return results.plot()
 
 gr.Interface(
-    fn=detect,
+    fn=predict,
     inputs=gr.Image(type="numpy"),
     outputs=gr.Image(type="numpy"),
-    title="Vehicle Detection"
+    title="Vehicle Detection App"
 ).launch()