animrods commited on
Commit
099fb5c
·
verified ·
1 Parent(s): 04ed389

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -24,14 +24,14 @@ def predict(prompt, ip_adapter_image, ip_adapter_scale=0.5, negative_prompt="",
24
  seed = random.randint(0, MAX_SEED)
25
 
26
  if not center_crop:
27
- ip_adapter_image = ip_adapter_image.resize((224,224))
28
 
29
  generator = torch.Generator(device="cuda").manual_seed(seed)
30
  pipe.set_ip_adapter_scale([ip_adapter_scale])
31
 
32
  image = pipe(
33
  prompt=prompt,
34
- ip_adapter_image=[ip_adapter_image],
35
  negative_prompt=negative_prompt,
36
  height=height,
37
  width=width,
@@ -68,7 +68,10 @@ with gr.Blocks(css=css) as demo:
68
 
69
  with gr.Row():
70
  with gr.Column():
71
- ip_adapter_image = gr.Image(label="IP-Adapter Image", type="pil")
 
 
 
72
  ip_adapter_scale = gr.Slider(
73
  label="Image Input Scale",
74
  info="Use 1 for creating image variations",
 
24
  seed = random.randint(0, MAX_SEED)
25
 
26
  if not center_crop:
27
+ ip_adapter_image = [x.resize((224,224)) for x in ip_adapter_image]
28
 
29
  generator = torch.Generator(device="cuda").manual_seed(seed)
30
  pipe.set_ip_adapter_scale([ip_adapter_scale])
31
 
32
  image = pipe(
33
  prompt=prompt,
34
+ ip_adapter_image=ip_adapter_image,
35
  negative_prompt=negative_prompt,
36
  height=height,
37
  width=width,
 
68
 
69
  with gr.Row():
70
  with gr.Column():
71
+ # ip_adapter_image = gr.Image(label="IP-Adapter Image", type="pil")
72
+ ip_adapter_image = gr.Gallery(
73
+ label="IP-Adapter Image/s", show_label=False, elem_id="gallery",
74
+ columns=[3], rows=[1], object_fit="contain", height="auto")
75
  ip_adapter_scale = gr.Slider(
76
  label="Image Input Scale",
77
  info="Use 1 for creating image variations",