Update app.py

app.py

@@ -69,22 +69,22 @@ def process_text_in_chunks(text, process_function, max_chunk_size=256):
     return processed_text.strip()
 @spaces.GPU(duration=120)
 def greet(img, apply_grammar_correction, apply_spell_check):
-    # img.save("out.jpg")
-    # doc = DocumentFile.from_images("out.jpg")
-    # output = OCRpredictor(doc)
+    img.save("out.jpg")
+    doc = DocumentFile.from_images("out.jpg")
+    output = OCRpredictor(doc)
 
-    # res = ""
-    # for obj in output.pages:
-    #     for obj1 in obj.blocks:
-    #         for obj2 in obj1.lines:
-    #             for obj3 in obj2.words:
-    #                 res += " " + obj3.value
-    #         res += "\n"
-    #     res += "\n"
+    res = ""
+    for obj in output.pages:
+        for obj1 in obj.blocks:
+            for obj2 in obj1.lines:
+                for obj3 in obj2.words:
+                    res += " " + obj3.value
+            res += "\n"
+        res += "\n"
 
 
     # img = cv2.imread(inputPath)
-    res = pt.image_to_string(img,lang='eng')
+    # res = pt.image_to_string(img,lang='eng')
     # print(text)
         
     # Process in chunks for grammar correction
@@ -104,6 +104,7 @@ demo_ocr = gr.Interface(
     fn=greet,
     inputs=[
         gr.Image(type="pil"),
+        gr.Dropdown(["English","Hindi","Punjabi"],label="Select Language"),
         gr.Checkbox(label="Apply Grammar Correction"),
         gr.Checkbox(label="Apply Spell Check")
     ],