Update app.py

app.py

@@ -87,18 +87,23 @@ description = "Select the model to use for generating the image description. 'Ba
 if device == "cpu":
     description += " Note: Running on CPU, which may be slow for large models."
 
-# Create the Gradio interface
-image_description_interface = gr.Interface(
-    fn=describe_image,
-    inputs=[
-        gr.Image(label="Upload Image", type="pil"),
-        gr.Radio(["Base", "Large"], label="Model Choice", value="Base")
-    ],
-    outputs=gr.Textbox(label="Generated Caption", lines=4, show_copy_button=True),
-    live=False,
-    title="Florence-2 Models Image Captions",
-    description=description
-)
+# Define examples
+examples = [
+    ["images/1.png", "Base"],
+    ["images/1.png", "Large"]
+]
+
+# Create the Gradio interface with Blocks
+with gr.Blocks(theme="bethecloud/storj_theme") as demo:
+    gr.Markdown("# **Florence-2 Models Image Captions**")
+    gr.Markdown(description)
+    with gr.Row():
+        image_input = gr.Image(label="Upload Image", type="pil")
+        model_choice = gr.Radio(["Base", "Large"], label="Model Choice", value="Base")
+    output = gr.Textbox(label="Generated Caption", lines=4, show_copy_button=True)
+    generate_btn = gr.Button("Generate Caption")
+    generate_btn.click(fn=describe_image, inputs=[image_input, model_choice], outputs=output)
+    gr.Examples(examples=examples, inputs=[image_input, model_choice])
 
 # Launch the interface
-image_description_interface.launch(debug=True)
+demo.launch(debug=True)