Update app.py

app.py

@@ -190,11 +190,14 @@ def create_gradio_interface():
             # Left Column (Inputs)
             with gr.Column(scale=1):
                 model_choice = gr.Dropdown(
-                    choices=["Camel-Doc-OCR-062825", "MonkeyOCR-Recognition", "Nanonets-OCR-s", "Megalodon-OCR-Sync-0713"],
+                    choices=["Camel-Doc-OCR-062825", 
+                             "MonkeyOCR-Recognition", 
+                             "Nanonets-OCR-s", 
+                             "Megalodon-OCR-Sync-0713"],
                     label="Select Model", value="Camel-Doc-OCR-062825"
                 )
                 task_choice = gr.Dropdown(
-                    choices=["Content Extraction", "Layout Analysis"],
+                    choices=["Content Extraction", "Layout Analysis(.json)"],
                     label="Select Task", value="Content Extraction"
                 )
                 image_input = gr.Image(label="Upload Image", type="pil", sources=['upload'])
@@ -223,12 +226,20 @@ def create_gradio_interface():
 
         process_btn.click(
             fn=process_document_stream,
-            inputs=[model_choice, task_choice, image_input, max_new_tokens],
-            outputs=[raw_output_stream, markdown_output, json_output]
+            inputs=[model_choice, 
+                    task_choice, 
+                    image_input, 
+                    max_new_tokens],
+            outputs=[raw_output_stream, 
+                     markdown_output, 
+                     json_output]
         )
         clear_btn.click(
             clear_all_outputs,
-            outputs=[image_input, raw_output_stream, markdown_output, json_output]
+            outputs=[image_input, 
+                     raw_output_stream, 
+                     markdown_output, 
+                     json_output]
         )
     return demo