Update app.py

app.py

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