Update app_blocks.py

app_blocks.py

@@ -5,10 +5,10 @@ from PIL import Image
 
 import gradio as gr
 
-def tesseract_ocr(filepath: str, languages: List[str]=None):
+def tesseract_ocr(filepath: str, languages: List[str]=None, psm: int = 7):
     image = Image.open(filepath)
     # return pytesseract.image_to_string(image=image, lang=', '.join(languages) if languages else None, config='--psm 7')
-    return pytesseract.image_to_string(image=image, lang=', '.join(languages))
+    return pytesseract.image_to_string(image=image, lang=', '.join(languages) if languages else None, config='--psm' + psm)
 
 title = "Tesseract OCR"
 description = "Gradio demo for Tesseract. Tesseract is an open source text recognition (OCR) Engine."
@@ -29,13 +29,14 @@ with gr.Blocks(title=title) as demo:
             language_choices = pytesseract.get_languages()
             with gr.Accordion("Languages", open=False):
                 languages = gr.CheckboxGroup(language_choices, type="value", value=["eng"], label='language')
+            psm_slider = gr.Slider(minimum=0, maximum=13, step=1, value=7, label="PSM level") 
             with gr.Row():
                 btn_clear = gr.ClearButton([image, languages])
                 btn_submit = gr.Button(value="Submit", variant="primary")
         with gr.Column():
             text = gr.Textbox(label="Output")
 
-    btn_submit.click(tesseract_ocr, inputs=[image, languages], outputs=text, api_name="tesseract-ocr")
+    btn_submit.click(tesseract_ocr, inputs=[image, languages, psm_slider], outputs=text, api_name="tesseract-ocr")
     btn_clear.add(text)
 
     gr.Examples(