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