Update app.py

app.py

@@ -28,7 +28,6 @@ def process_image(prompt, image, style, upscale_factor, inpaint):
         print(f"Inpaint: {inpaint}")
 
         # Example placeholder logic for using the pipeline
-        # Replace with actual pipeline logic based on selected style
         if inpaint and inpaint_model:
             result = inpaint_model(prompt=prompt, image=image, guidance_scale=7.5)
         else:
@@ -55,7 +54,7 @@ with gr.Blocks() as demo:
             upscale_input = gr.Slider(minimum=1, maximum=4, step=1, label="Upscale Factor")
             inpaint_input = gr.Checkbox(label="Enable Inpainting")
             
-            output = gr.Image(label="Generated Image")
+            output = gr.Image(label="Generated Image", type="pil")
 
             generate_button = gr.Button("Generate Image")
             generate_button.click(