Update app.py

app.py

@@ -36,6 +36,9 @@ default_images = {
     "Vest": "dress.jpg"
 }
 
+# Convert image paths to a format suitable for Gradio
+default_image_paths = [[img_path] for img_path in default_images.values()]
+
 # **Image Preprocessing Function**
 def preprocess_image(image):
     """Applies necessary transformations to the input image."""
@@ -104,23 +107,31 @@ with gr.Blocks() as interface:
     gr.Markdown("# Clothing1M Image Classifier")
     gr.Markdown("Upload a clothing image or select from the predefined images below.")
 
-    # Default Image Selection
+    # **Default Image Gallery for Selection**
+    gr.Markdown("### Select a Default Image:")
+    gallery = gr.Gallery(
+        value=default_image_paths,
+        label="Available Default Images",
+        show_label=True
+    )
+
+    # **Default Image Selection Radio**
     default_selector = gr.Radio(
         choices=list(default_images.keys()),
         label="Select a Default Image",
-        value="T-Shirt"
+        value="Shawl"
     )
 
-    # File Upload Option
+    # **File Upload Option**
     image_upload = gr.Image(type="numpy", label="Or Upload Your Own Image")
 
-    # Output Text
+    # **Output Text**
     output_text = gr.Textbox(label="Classification Result")
 
-    # Classify Button
+    # **Classify Button**
     classify_button = gr.Button("Classify Image")
 
-    # Define Action
+    # **Define Action**
     classify_button.click(
         fn=classify_image,
         inputs=[default_selector, image_upload],