SIGMitch commited on
Commit
5047bba
·
verified ·
1 Parent(s): 8d7a457

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -116,7 +116,7 @@ def process(
116
  image=resized_image,
117
  width=width,
118
  height=height,
119
- num_images_per_prompt =2,
120
  strength=strength_slider,
121
  generator=generator,
122
  joint_attention_kwargs={"scale": num_influence},
@@ -125,7 +125,7 @@ def process(
125
  )
126
  print('INFERENCE DONE')
127
  #return result.images[0].resize((image.size), Image.LANCZOS), result.images[1].resize((image.size), Image.LANCZOS)
128
- return result.images[0], result.images[1]
129
 
130
  #resized_mask = mask.resize((width, height), Image.LANCZOS)
131
 
 
116
  image=resized_image,
117
  width=width,
118
  height=height,
119
+ num_images_per_prompt =1,
120
  strength=strength_slider,
121
  generator=generator,
122
  joint_attention_kwargs={"scale": num_influence},
 
125
  )
126
  print('INFERENCE DONE')
127
  #return result.images[0].resize((image.size), Image.LANCZOS), result.images[1].resize((image.size), Image.LANCZOS)
128
+ return result.images[0], None #result.images[1]
129
 
130
  #resized_mask = mask.resize((width, height), Image.LANCZOS)
131