bghira commited on
Commit
8be7ccb
·
verified ·
1 Parent(s): c544475

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -18,13 +18,14 @@ def extract_resolution(resolution_str):
18
  return None
19
 
20
  # Define the image generation function with adjustable parameters and a progress bar
21
- @spaces.GPU
22
- def generate(prompt, guidance_scale, num_inference_steps, resolution, negative_prompt):
23
  width, height = extract_resolution(resolution) or (1024, 1024)
24
  return pipe(
25
  prompt,
26
  negative_prompt=negative_prompt,
27
  guidance_scale=guidance_scale,
 
28
  num_inference_steps=num_inference_steps,
29
  width=width, height=height
30
  ).images
@@ -41,7 +42,8 @@ iface = gr.Interface(
41
  fn=generate,
42
  inputs=[
43
  gr.Text(label="Enter your prompt"),
44
- gr.Slider(1, 20, step=0.1, label="Guidance Scale", value=5),
 
45
  gr.Slider(1, 50, step=1, label="Number of Inference Steps", value=35),
46
  gr.Radio(["1024x1024", "1152x960", "896x1152"], label="Resolution", value="1152x960"),
47
  gr.Text(value="underexposed, blurry, ugly, washed-out", label="Negative Prompt")
@@ -67,4 +69,3 @@ iface = gr.Interface(
67
  "<li>Datasets: @jimmycarter, @pseudoterminalx</li>"
68
  "</ul>"
69
  )
70
- ).launch()
 
18
  return None
19
 
20
  # Define the image generation function with adjustable parameters and a progress bar
21
+ @spaces.GPU(duration=115)
22
+ def generate(prompt, guidance_scale, guidance_scale_real, num_inference_steps, resolution, negative_prompt):
23
  width, height = extract_resolution(resolution) or (1024, 1024)
24
  return pipe(
25
  prompt,
26
  negative_prompt=negative_prompt,
27
  guidance_scale=guidance_scale,
28
+ guidance_scale_real=guidance_scale_real,
29
  num_inference_steps=num_inference_steps,
30
  width=width, height=height
31
  ).images
 
42
  fn=generate,
43
  inputs=[
44
  gr.Text(label="Enter your prompt"),
45
+ gr.Slider(1, 4, step=0.1, label="Guidance Scale (Flux)", value=3.5),
46
+ gr.Slider(1, 12, step=0.1, label="Guidance Scale (CFG)", value=5.0),
47
  gr.Slider(1, 50, step=1, label="Number of Inference Steps", value=35),
48
  gr.Radio(["1024x1024", "1152x960", "896x1152"], label="Resolution", value="1152x960"),
49
  gr.Text(value="underexposed, blurry, ugly, washed-out", label="Negative Prompt")
 
69
  "<li>Datasets: @jimmycarter, @pseudoterminalx</li>"
70
  "</ul>"
71
  )