fix: image pipeline arguments

app.py

@@ -22,6 +22,7 @@ def infer(prompt, init_image=None, seed=42, randomize_seed=False, width=1024, he
     
     if init_image is not None:
         # Process img2img
+        init_image = init_image.convert("RGB")
         init_image = pipe.preprocess(init_image).unsqueeze(0).to(device, dtype)
         image = pipe(
             prompt=prompt,
@@ -108,7 +109,7 @@ with gr.Blocks(css=css) as demo:
             run_button = gr.Button("Run", scale=0)
 
         with gr.Row():
-            init_image = gr.Image(label="Initial Image (optional)", type="pil", optional=True)
+            init_image = gr.Image(label="Initial Image (optional)", type="pil")
             result = gr.Image(label="Result", show_label=False)
 
         with gr.Accordion("Advanced Settings", open=False):
@@ -162,3 +163,4 @@ with gr.Blocks(css=css) as demo:
     )
 
 demo.launch()
+

requirements.txt

@@ -4,4 +4,5 @@ invisible_watermark
 torch
 transformers==4.42.4
 xformers
-sentencepiece
+sentencepiece
+gradio==4.29.0