DGSpitzer commited on
Commit
46810b9
1 Parent(s): fbaf487

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -54,7 +54,7 @@ def inference(prompt, negPrompt, guidance, steps, width_input, height_input):
54
  def inference_example(prompt, negPrompt, guidance, steps, width_input, height_input):
55
 
56
  prompt = prompt_prefixes[current_model] + prompt
57
- results = pipe(prompt, negative_prompt, num_inference_steps=int(steps), guidance_scale=guidance, width=width_input, height=height_input)
58
  image = results.images[0] if not results.nsfw_content_detected[0] else Image.open("nsfw_placeholder.jpg")
59
  return image
60
 
 
54
  def inference_example(prompt, negPrompt, guidance, steps, width_input, height_input):
55
 
56
  prompt = prompt_prefixes[current_model] + prompt
57
+ results = pipe(prompt, negative_prompt = negPrompt, num_inference_steps=int(steps), guidance_scale=guidance, width=width_input, height=height_input)
58
  image = results.images[0] if not results.nsfw_content_detected[0] else Image.open("nsfw_placeholder.jpg")
59
  return image
60