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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -44,7 +44,9 @@ def on_model_change(model):
44
  pipe = pipe.to("cuda")
45
 
46
  def inference(prompt, negPrompt, guidance, steps, width_input, height_input):
47
-
 
 
48
  promptPrev = prompt
49
  prompt = prompt_prefixes[current_model] + prompt
50
  results = pipe(prompt, negative_prompt = negPrompt, num_inference_steps=int(steps), guidance_scale=guidance, width=width_input, height=height_input)
@@ -52,7 +54,7 @@ def inference(prompt, negPrompt, guidance, steps, width_input, height_input):
52
  return image, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True), gr.update(placeholder=promptPrev)
53
 
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")
 
44
  pipe = pipe.to("cuda")
45
 
46
  def inference(prompt, negPrompt, guidance, steps, width_input, height_input):
47
+ print("Generated image with prompt: " + prompt)
48
+ if negPrompt != "":
49
+ print("Negative prompt: " + negPrompt)
50
  promptPrev = prompt
51
  prompt = prompt_prefixes[current_model] + prompt
52
  results = pipe(prompt, negative_prompt = negPrompt, num_inference_steps=int(steps), guidance_scale=guidance, width=width_input, height=height_input)
 
54
  return image, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True), gr.update(placeholder=promptPrev)
55
 
56
  def inference_example(prompt, negPrompt, guidance, steps, width_input, height_input):
57
+
58
  prompt = prompt_prefixes[current_model] + prompt
59
  results = pipe(prompt, negative_prompt = negPrompt, num_inference_steps=int(steps), guidance_scale=guidance, width=width_input, height=height_input)
60
  image = results.images[0] if not results.nsfw_content_detected[0] else Image.open("nsfw_placeholder.jpg")