Update app.py

app.py

@@ -46,12 +46,12 @@ def describe_image(uploaded_image, model_choice):
     if uploaded_image is None:
         return "Please upload an image."
 
-    if model_choice == "Base":
+    if model_choice == "Florence-2-base":
         if vision_language_model_base is None:
             return "Base model failed to load."
         model = vision_language_model_base
         processor = vision_language_processor_base
-    elif model_choice == "Large":
+    elif model_choice == "Florence-2-large":
         if vision_language_model_large is None:
             return "Large model failed to load."
         model = vision_language_model_large
@@ -89,10 +89,10 @@ if device == "cpu":
 
 # Define examples
 examples = [
-    ["images/1.png", "Base"],
-    ["images/1.png", "Large"],
-    ["images/2.png", "Base"],
-    ["images/2.png", "Large"]
+    ["images/1.png", "Florence-2-base"],
+    ["images/1.png", "Florence-2-large"],
+    ["images/2.png", "Florence-2-base"],
+    ["images/2.png", "Florence-2-large"]
 ]
 
 css = """
@@ -114,15 +114,12 @@ with gr.Blocks(theme="bethecloud/storj_theme", css=css) as demo:
         with gr.Column():
             image_input = gr.Image(label="Upload Image", type="pil")
             generate_btn = gr.Button("Generate Caption", elem_classes="submit-btn")
+            gr.Examples(examples=examples, inputs=[image_input, model_choice])
         # Right column: Model choice, output, and examples
         with gr.Column():
             model_choice = gr.Radio(["Base", "Large"], label="Model Choice", value="Base")
-
             with gr.Row():
                 output = gr.Textbox(label="Generated Caption", lines=4, show_copy_button=True)
-            
-            gr.Examples(examples=examples, inputs=[image_input, model_choice])
-
     # Connect the button to the function
     generate_btn.click(fn=describe_image, inputs=[image_input, model_choice], outputs=output)