DarianT commited on
Commit
27c1ccc
·
1 Parent(s): 6667420

Update advanced settings

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -32,6 +32,7 @@ def infer(
32
  guidance_scale,
33
  num_inference_steps,
34
  progress=gr.Progress(track_tqdm=True),
 
35
  ):
36
  if randomize_seed:
37
  seed = random.randint(0, MAX_SEED)
@@ -43,7 +44,10 @@ def infer(
43
 
44
  # Construct prompt from dropdown selections
45
  prompt = f"face {pose.lower()} photo of {gender} {id} person, {background.lower()} background"
46
- ""
 
 
 
47
  image = pipe(
48
  prompt=prompt,
49
  negative_prompt=negative_prompt,
@@ -52,6 +56,7 @@ def infer(
52
  width=image_size,
53
  height=image_size,
54
  generator=generator,
 
55
  ).images[0]
56
 
57
  return image, seed
@@ -95,7 +100,7 @@ with gr.Blocks(css=css) as demo:
95
  label="Negative prompt",
96
  max_lines=1,
97
  placeholder="Enter a negative prompt",
98
- visible=False,
99
  )
100
  num_inference_steps = gr.Slider(
101
  label="Number of sample steps",
@@ -146,6 +151,7 @@ with gr.Blocks(css=css) as demo:
146
  randomize_seed,
147
  guidance_scale,
148
  num_inference_steps,
 
149
  ],
150
  outputs=[result, seed],
151
  )
 
32
  guidance_scale,
33
  num_inference_steps,
34
  progress=gr.Progress(track_tqdm=True),
35
+ num_images
36
  ):
37
  if randomize_seed:
38
  seed = random.randint(0, MAX_SEED)
 
44
 
45
  # Construct prompt from dropdown selections
46
  prompt = f"face {pose.lower()} photo of {gender} {id} person, {background.lower()} background"
47
+
48
+ print(prompt)
49
+ print(negative_prompt)
50
+
51
  image = pipe(
52
  prompt=prompt,
53
  negative_prompt=negative_prompt,
 
56
  width=image_size,
57
  height=image_size,
58
  generator=generator,
59
+ num_images_per_prompt=num_images,
60
  ).images[0]
61
 
62
  return image, seed
 
100
  label="Negative prompt",
101
  max_lines=1,
102
  placeholder="Enter a negative prompt",
103
+ value="cartoon, cgi, render, illustration, painting, drawing, black and white, bad body proportions, landscape",
104
  )
105
  num_inference_steps = gr.Slider(
106
  label="Number of sample steps",
 
151
  randomize_seed,
152
  guidance_scale,
153
  num_inference_steps,
154
+ num_images
155
  ],
156
  outputs=[result, seed],
157
  )