falba commited on
Commit
c49631d
1 Parent(s): 598a04f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ qreader = QReader()
17
  def yolo_and_trocr(image_input, save):
18
  try:
19
  # YOLO instanciated from the trained model
20
- yolo = YOLO('/content/best.pt')
21
 
22
  # Creating results
23
  results = yolo(image_input, conf=0.5, iou=0.7)
 
17
  def yolo_and_trocr(image_input, save):
18
  try:
19
  # YOLO instanciated from the trained model
20
+ yolo = YOLO('best.pt')
21
 
22
  # Creating results
23
  results = yolo(image_input, conf=0.5, iou=0.7)