amos1088 commited on
Commit
bca93ec
·
1 Parent(s): 36e35d5
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -59,8 +59,8 @@ def gui_generation(prompt, ref_img, guidance_scale, ipadapter_scale):
59
  guidance_scale=guidance_scale,
60
  generator=torch.Generator("cuda").manual_seed(42),
61
  clip_image=ref_img.convert('RGB'),
62
- ipadapter_scale=ipadapter_scale).images[0]
63
- return image
64
 
65
  # Create Gradio interface
66
  with gr.Blocks() as demo:
 
59
  guidance_scale=guidance_scale,
60
  generator=torch.Generator("cuda").manual_seed(42),
61
  clip_image=ref_img.convert('RGB'),
62
+ ipadapter_scale=ipadapter_scale).images
63
+ return image[0]
64
 
65
  # Create Gradio interface
66
  with gr.Blocks() as demo: