artbreguez commited on
Commit
204de19
·
verified ·
1 Parent(s): 7886d5a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from PIL import Image
6
  import gradio as gr
7
  import numpy as np
8
 
9
- model = torch.hub.load('ultralytics/yolov5', 'custom', path='yolo-v5.pt')
10
  model.conf = 0.80
11
 
12
  processor = TrOCRProcessor.from_pretrained('microsoft/trocr-base-printed')
 
6
  import gradio as gr
7
  import numpy as np
8
 
9
+ model = torch.hub.load('ultralytics/yolov5', 'custom', path='./yolo-v5.pt')
10
  model.conf = 0.80
11
 
12
  processor = TrOCRProcessor.from_pretrained('microsoft/trocr-base-printed')