robin0307 commited on
Commit
07daf83
1 Parent(s): e7d6df7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -25,8 +25,8 @@ def inference(img, det, recog):
25
  recog = None
26
  ocr = MMOCR(det=det, recog=recog)
27
  print(datetime.datetime.now(), 'start read:', img.name)
28
- results = ocr.readtext(img.name, details=True, output='output')
29
- result_file = 'output/out_{}.png'.format(os.path.splitext(os.path.basename(img.name))[0])
30
  print(datetime.datetime.now(), results)
31
  # return result_file, pd.DataFrame(results[0]['result']).iloc[: , 2:]
32
  return result_file, results
 
25
  recog = None
26
  ocr = MMOCR(det=det, recog=recog)
27
  print(datetime.datetime.now(), 'start read:', img.name)
28
+ results = ocr.readtext(img.name, details=True, output='/tmp')
29
+ result_file = '/tmp/out_{}.png'.format(os.path.splitext(os.path.basename(img.name))[0])
30
  print(datetime.datetime.now(), results)
31
  # return result_file, pd.DataFrame(results[0]['result']).iloc[: , 2:]
32
  return result_file, results