Update app.py

app.py

@@ -89,16 +89,21 @@ examples = [
 ]
 
 # Gradio interface
-iface = gr.Interface(
-    fn=process_image,
-    inputs=[
-        gr.Image(type="pil", label="Input Image"),
-        gr.Dropdown(choices=list(MODEL_OPTIONS.keys()), label="Select Model", value="Medieval Base")
-    ],
-    outputs=gr.Textbox(label="Transcription"),
-    title="Medieval TrOCR Model Switcher",
-    description="Upload an image of medieval text and select a model to transcribe it. Note: This tool is designed to work on a single line of text at a time for optimal results.",
-    examples=examples
-)
+with gr.Blocks() as iface:
+    gr.Markdown("# Medieval TrOCR Model Switcher")
+    gr.Markdown("Upload an image of medieval text and select a model to transcribe it. Note: This tool is designed to work on a single line of text at a time for optimal results.")
+    
+    with gr.Row():
+        with gr.Column(scale=2):
+            input_image = gr.Image(type="pil", label="Input Image")
+        with gr.Column(scale=1):
+            model_dropdown = gr.Dropdown(choices=list(MODEL_OPTIONS.keys()), label="Select Model", value="Medieval Base")
+    
+    transcription_output = gr.Textbox(label="Transcription")
+    
+    submit_button = gr.Button("Transcribe")
+    submit_button.click(fn=process_image, inputs=[input_image, model_dropdown], outputs=transcription_output)
+    
+    gr.Examples(examples, inputs=[input_image, model_dropdown], outputs=transcription_output)
 
 iface.launch()