srimanth-d commited on
Commit
19abc97
·
verified ·
1 Parent(s): 33786b1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ def extract_text(image_bytes):
32
  image.save("temp_image.png", format="PNG")
33
 
34
  # Extract text using the cached model
35
- res = model.chat(tokenizer, "temp_image.png", ocr_type='format')
36
  return res
37
 
38
  except Exception as e:
 
32
  image.save("temp_image.png", format="PNG")
33
 
34
  # Extract text using the cached model
35
+ res = model.chat(tokenizer, "temp_image.png", ocr_type='ocr')
36
  return res
37
 
38
  except Exception as e: