Update app.py

app.py

@@ -88,16 +88,31 @@ examples = [
     [f"{base_url}textualis-3.png", "Medieval Textualis"],
 ]
 
+# Custom CSS to make the image wider
+custom_css = """
+#image_upload {
+    max-width: 100% !important;
+    width: 100% !important;
+    height: auto !important;
+}
+#image_upload > div:first-child {
+    width: 100% !important;
+}
+#image_upload img {
+    max-width: 100% !important;
+    width: 100% !important;
+    height: auto !important;
+}
+"""
+
 # Gradio interface
-with gr.Blocks() as iface:
+with gr.Blocks(css=custom_css) 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")
+        input_image = gr.Image(type="pil", label="Input Image", elem_id="image_upload")
+        model_dropdown = gr.Dropdown(choices=list(MODEL_OPTIONS.keys()), label="Select Model", value="Medieval Base")
     
     transcription_output = gr.Textbox(label="Transcription")