Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
|