fffiloni commited on
Commit
84d7069
·
1 Parent(s): 5fabe6a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ gallery = gr.Gallery(label="Generated images", show_label=False, elem_id="galler
25
 
26
  def infer(prompt, init_image):
27
  print(init_image)
28
- init_image = Image.open(BytesIO(init_image)).convert("RGB")
29
  init_image = init_image.resize((768, 512))
30
  #image = pipe(prompt, init_image=init_image)["sample"][0]
31
  images_list = pipe([prompt] * 2, init_image=init_image, strength=0.75)
 
25
 
26
  def infer(prompt, init_image):
27
  print(init_image)
28
+ init_image = Image.open(init_image)
29
  init_image = init_image.resize((768, 512))
30
  #image = pipe(prompt, init_image=init_image)["sample"][0]
31
  images_list = pipe([prompt] * 2, init_image=init_image, strength=0.75)