Update app_with_diffusers.py

app_with_diffusers.py

@@ -39,13 +39,13 @@ pipe.aggregator.load_state_dict(pretrained_state_dict)
 pipe.to(device='cuda', dtype=torch.float16)
 pipe.aggregator.to(device='cuda', dtype=torch.float16)
 
-def infer(input_image):
+def infer(prompt, input_image):
     # load a broken image
     low_quality_image = Image.open(input_image).convert("RGB")
     
     # InstantIR restoration
     image = pipe(
-        prompt="",
+        prompt=prompt,
         image=low_quality_image,
         previewer_scheduler=lcm_scheduler,
     ).images[0]
@@ -59,11 +59,12 @@ with gr.Blocks() as demo:
         with gr.Row():
             with gr.Column():
                 lq_img = gr.Image(label="Low-quality image", type="filepath")
+                prompt = gr.Textbox(label="Prompt", value="")
                 submit_btn = gr.Button("InstantIR magic!")
             output_img = gr.Image(label="InstantIR restored")
     submit_btn.click(
         fn=infer,
-        inputs=[lq_img],
+        inputs=[prompt, lq_img],
         outputs=[output_img]
     )
 demo.launch(show_error=True)