Update app.py

app.py

@@ -68,7 +68,21 @@ 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):
+def greet(img, apply_grammar_correction, apply_spell_check,lang_of_input):
+
+    if (lang_of_input=="Hindi"):
+        res = pt.image_to_string(img,lang='hin')
+        _output_name = "RESULT_OCR.txt"
+        open(_output_name, 'w').write(res)
+        return res, _output_name
+
+    if (lang_of_input=="Punjabi"):
+        res = pt.image_to_string(img,lang='pan')
+        _output_name = "RESULT_OCR.txt"
+        open(_output_name, 'w').write(res)
+        return res, _output_name
+       
+        
     img.save("out.jpg")
     doc = DocumentFile.from_images("out.jpg")
     output = OCRpredictor(doc)
@@ -81,11 +95,6 @@ def greet(img, apply_grammar_correction, apply_spell_check):
                     res += " " + obj3.value
             res += "\n"
         res += "\n"
-
-
-    # img = cv2.imread(inputPath)
-    # res = pt.image_to_string(img,lang='eng')
-    # print(text)
         
     # Process in chunks for grammar correction
     if apply_grammar_correction:
@@ -104,9 +113,9 @@ 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")
+        gr.Checkbox(label="Apply Spell Check"),
+        gr.Dropdown(["English","Hindi","Punjabi"],label="Select Language")
     ],
     outputs=["text", "file"],
     title="DocTR OCR with Grammar and Spell Check",