Update app.py

app.py

@@ -148,8 +148,8 @@ def infer(input_image, prompt, illumination_dropdown, direction_dropdown, seed=4
         progress (gr.Progress): A Gradio progress tracker for the UI.
     
     Returns:
-        tuple[PIL.Image.Image, int, str]: A tuple containing:
-            - The relighted output image.
+        tuple[list[PIL.Image.Image], int, str]: A tuple containing:
+            - A list with the input image (image[0]) and the relighted output image (image[1]).
             - The seed used for the generation.
             - The final constructed prompt string used by the model.