uuu

app.py

@@ -21,7 +21,7 @@ import gradio as gr
 import torch
 from pipeline_stable_diffusion_3_ipa import StableDiffusion3Pipeline
 import os
-import numpy as np
+from PIL import Image
 import spaces
 from huggingface_hub import login
 token = os.getenv("HF_TOKEN")
@@ -48,6 +48,8 @@ pipe.init_ipadapter(
 
 @spaces.GPU
 def gui_generation(prompt, ref_img, guidance_scale, ipadapter_scale):
+    ref_img = Image.open(ref_img).convert('RGB')
+
     with torch.no_grad():
         # Ensure the pipeline runs with correct dtype and device
         image = pipe(
@@ -70,7 +72,7 @@ with gr.Blocks() as demo:
         prompt_box = gr.Textbox(label="Prompt", placeholder="Enter your image generation prompt")
 
     with gr.Row():
-        ref_img = gr.Image(type="pil", label="Upload Reference Image")
+        ref_img = gr.Image(type="filepath", label="Upload Reference Image")
 
     with gr.Row():
         guidance_slider = gr.Slider(