anzorq commited on
Commit
2cc95c9
1 Parent(s): 0e5eebf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -8
app.py CHANGED
@@ -127,9 +127,9 @@ def inference(model_name, prompt, guidance, steps, n_images=1, width=512, height
127
 
128
  try:
129
  if img is not None:
130
- return img_to_img(model_path, prompt, n_images, neg_prompt, img, strength, guidance, steps, width, height, generator, seed), f"Done. Seed: {seed}"
131
  else:
132
- return txt_to_img(model_path, prompt, n_images, neg_prompt, guidance, steps, width, height, generator, seed), f"Done. Seed: {seed}"
133
  except Exception as e:
134
  return None, error_str(e)
135
 
@@ -181,7 +181,7 @@ def inference_image(model_path, prompt, mode, *args, **kwargs):
181
  )
182
  return replace_nsfw_images(result)
183
 
184
- def txt_to_img(model_path, prompt, n_images, neg_prompt, guidance, steps, width, height, generator, seed):
185
  return inference_image(model_path, prompt, "txt2img",
186
  negative_prompt=neg_prompt,
187
  num_images_per_prompt=n_images,
@@ -189,11 +189,10 @@ def txt_to_img(model_path, prompt, n_images, neg_prompt, guidance, steps, width,
189
  guidance_scale=guidance,
190
  width=width,
191
  height=height,
192
- generator=generator,
193
- seed=seed
194
  )
195
 
196
- def img_to_img(model_path, prompt, n_images, neg_prompt, img, strength, guidance, steps, width, height, generator, seed):
197
  img = process_image(img, width, height)
198
  return inference_image(model_path, prompt, "img2img",
199
  negative_prompt=neg_prompt,
@@ -202,8 +201,7 @@ def img_to_img(model_path, prompt, n_images, neg_prompt, img, strength, guidance
202
  num_inference_steps=int(steps),
203
  strength=strength,
204
  guidance_scale=guidance,
205
- generator=generator,
206
- seed=seed
207
  )
208
 
209
  def replace_nsfw_images(results):
 
127
 
128
  try:
129
  if img is not None:
130
+ return img_to_img(model_path, prompt, n_images, neg_prompt, img, strength, guidance, steps, width, height, generator), f"Done. Seed: {seed}"
131
  else:
132
+ return txt_to_img(model_path, prompt, n_images, neg_prompt, guidance, steps, width, height, generator), f"Done. Seed: {seed}"
133
  except Exception as e:
134
  return None, error_str(e)
135
 
 
181
  )
182
  return replace_nsfw_images(result)
183
 
184
+ def txt_to_img(model_path, prompt, n_images, neg_prompt, guidance, steps, width, height, generator):
185
  return inference_image(model_path, prompt, "txt2img",
186
  negative_prompt=neg_prompt,
187
  num_images_per_prompt=n_images,
 
189
  guidance_scale=guidance,
190
  width=width,
191
  height=height,
192
+ generator=generator
 
193
  )
194
 
195
+ def img_to_img(model_path, prompt, n_images, neg_prompt, img, strength, guidance, steps, width, height, generator):
196
  img = process_image(img, width, height)
197
  return inference_image(model_path, prompt, "img2img",
198
  negative_prompt=neg_prompt,
 
201
  num_inference_steps=int(steps),
202
  strength=strength,
203
  guidance_scale=guidance,
204
+ generator=generator
 
205
  )
206
 
207
  def replace_nsfw_images(results):