uuu

app.py

@@ -24,6 +24,8 @@ import os
 from PIL import Image
 import spaces
 from huggingface_hub import login
+from diffusers.utils import load_image
+
 token = os.getenv("HF_TOKEN")
 login(token=token)
 
@@ -48,7 +50,7 @@ pipe.init_ipadapter(
 
 @spaces.GPU
 def gui_generation(prompt, ref_img, guidance_scale, ipadapter_scale):
-    ref_img = Image.open(ref_img).convert('RGB')
+    ref_img = load_image(ref_img.name)
 
     with torch.no_grad():
         # Ensure the pipeline runs with correct dtype and device
@@ -93,7 +95,7 @@ with gr.Blocks() as demo:
         )
 
     generate_btn = gr.Button("Generate")
-    gallery = gr.Image(type="pil", label="Generated Image")
+    gallery = gr.File(type="pil", label="Generated Image")
 
     generate_btn.click(
         fn=gui_generation,