Update app.py

app.py

@@ -31,10 +31,8 @@ def modify_caption(caption: str) -> str:
 def create_captions_rich(images):
     """
     Generates captions for input images.
-
     Args:
         images (list): List of images to generate captions for.
-
     Returns:
         list: List of captions, one for each input image.
     """
@@ -69,10 +67,10 @@ with gr.Blocks(css=css) as demo:
   with gr.Tab(label="Image to Prompt for SD3"):
     with gr.Row():
       with gr.Column():
-        input_img = gr.Image(label="Input Image", type="pil", interactive=True,  multiple=True)
+        input_img = gr.Gallery(label="Input Images", type="pil", interactive=True)
         submit_btn = gr.Button(value="Start")
       output = gr.Textbox(label="Prompt", lines=10, interactive=True)
 
     submit_btn.click(create_captions_rich, [input_img], [output])
 
-demo.launch(debug=True)
+demo.launch(debug=True)