uuu

app.py

@@ -65,14 +65,6 @@ pipe.init_ipadapter(
 
 @spaces.GPU
 def gui_generation(prompt, ref_img, guidance_scale, ipadapter_scale):
-    # Ensure reference image is in the correct format
-    if ref_img:
-        ref_img = ref_img.convert("RGB")
-        ref_img_tensor = torch.tensor(
-            np.array(ref_img), dtype=torch.bfloat16, device="cuda"
-        )
-    else:
-        raise ValueError("Reference image is required.")
 
     # Ensure the pipeline runs with correct dtype and device
     with torch.autocast("cuda", dtype=torch.bfloat16):
@@ -84,7 +76,7 @@ def gui_generation(prompt, ref_img, guidance_scale, ipadapter_scale):
             num_inference_steps=24,
             guidance_scale=guidance_scale,
             generator=torch.Generator("cuda").manual_seed(42),
-            clip_image=ref_img_tensor,
+            clip_image=ref_img,
             ipadapter_scale=ipadapter_scale,
         ).images[0]