ui change

app.py

@@ -26,12 +26,23 @@ def generate_caption(image):
 
     return generated_caption
 
+# # Examples for testing
+# examples = [
+#     ["example_images/image1.jpg"],  # Replace with paths to example images
+#     ["example_images/image2.jpg"]
+# ]
+
 # Define the Gradio interface
 interface = gr.Interface(
     fn=generate_caption,  # Function to process input
-    inputs=gr.Image(),    # Input as image
-    outputs=gr.Textbox(), # Output as text
-    live=True             # Enable live prediction
+    inputs=gr.Image(label="Upload an Image"),  # Add a label to input
+    outputs=gr.Textbox(label="Generated Caption", lines=2),  # Larger textbox for output
+    # examples=examples,  # Add example images
+    live=True,  # Enable live prediction
+    title="📸 Image Caption Generator",  # Add a title
+    description="Upload an chest x-ray image to generate a descriptive caption using our AI model. Built with Transformers and Gradio.",  # Add a description
+    theme="allenai/gradio-theme",  # Use Gradio's built-in themes
+    css=".output { font-size: 16px; padding: 10px; border: 1px solid #ccc; border-radius: 5px; }",  # Custom CSS for output styling
 )
 
 # Launch the Gradio app