Update app.py

app.py

@@ -5,10 +5,10 @@ import re
 from PIL import Image
 import gradio as gr
 import numpy as np
-from ultralytics import YOLO
+import yolov5
+
+model = yolov5.load('yolo-v5.pt')
 
-# model = torch.hub.load('ultralytics/yolov5', 'custom', path='./yolo-v5.pt')
-model = YOLO('yolo-v5.pt')
 
 model.conf = 0.80