Spaces:

srimanth-d
/

OCR_app

Sleeping

srimanth-d commited on Apr 27

Commit

19abc97

verified ·

1 Parent(s): 33786b1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -32,7 +32,7 @@ def extract_text(image_bytes):
         image.save("temp_image.png", format="PNG")
         # Extract text using the cached model
-        res = model.chat(tokenizer, "temp_image.png", ocr_type='format')
         return res
     except Exception as e:

         image.save("temp_image.png", format="PNG")
         # Extract text using the cached model
+        res = model.chat(tokenizer, "temp_image.png", ocr_type='ocr')
         return res
     except Exception as e: