Update app.py

app.py

@@ -5,8 +5,11 @@ import re
 from PIL import Image
 import gradio as gr
 import numpy as np
+from ultralytics import YOLO
+
+# model = torch.hub.load('ultralytics/yolov5', 'custom', path='./yolo-v5.pt')
+model = YOLO('yolo-v5.pt')
 
-model = torch.hub.load('ultralytics/yolov5', 'custom', path='yolo-v5.pt')
 model.conf = 0.80 
 
 processor = TrOCRProcessor.from_pretrained('microsoft/trocr-base-printed')