surinala commited on
Commit
9128cbf
·
1 Parent(s): 68cbf33

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -137,7 +137,7 @@ def main():
137
  temp_file.write(uploaded_file.read())
138
  temp_file_path = temp_file.name
139
 
140
- doc = fitz.open(temp_file_path)
141
  predictor = ocr_predictor(pretrained=True)
142
  result = predictor(doc)
143
  json_export = result.export()
 
137
  temp_file.write(uploaded_file.read())
138
  temp_file_path = temp_file.name
139
 
140
+ doc = DocumentFile.from_pdf(temp_file_path)
141
  predictor = ocr_predictor(pretrained=True)
142
  result = predictor(doc)
143
  json_export = result.export()