Update app.py

app.py

@@ -4,7 +4,6 @@ import spaces
 import torch
 import re
 
-# Load the model and processor
 model = PaliGemmaForConditionalGeneration.from_pretrained("gokaygokay/sd3-long-captioner").to("cpu").eval()
 processor = PaliGemmaProcessor.from_pretrained("gokaygokay/sd3-long-captioner")
 
@@ -66,11 +65,11 @@ css = """
 """
 
 with gr.Blocks(css=css) as demo:
-  gr.HTML("<h1><center>Image caption using finetuned PaliGemma on SD3 generation data.<center><h1>")
-  with gr.Tab(label="Img2Prompt for SD3"):
+  gr.HTML("<h1><center>Finetuned PaliGemma for SD3 prompt generation.<center><h1>")
+  with gr.Tab(label="Image to Prompt for SD3"):
     with gr.Row():
       with gr.Column():
-        input_img = gr.Image(label="Input Image",  tool="select", type="pil",  interactive=True)
+        input_img = gr.Image(label="Input Image", type="pil", interactive=True,  multiple=True)
         submit_btn = gr.Button(value="Start")
       output = gr.Textbox(label="Prompt", lines=10, interactive=True)