Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -12,8 +12,7 @@ import zipfile
|
|
12 |
from transformers import CLIPTextModel, CLIPTokenizer, T5EncoderModel, T5TokenizerFast
|
13 |
|
14 |
# Description for the app
|
15 |
-
DESCRIPTION = """##
|
16 |
-
Choose between 'flux.1-dev-realism' for hyper-realistic images or 'flux.1-krea' for creative outputs."""
|
17 |
|
18 |
# Helper functions
|
19 |
def save_image(img):
|
@@ -363,7 +362,7 @@ def generate(
|
|
363 |
zip_images=zip_images,
|
364 |
progress=progress,
|
365 |
)
|
366 |
-
elif model_choice == "flux.1-krea":
|
367 |
return generate_krea(
|
368 |
prompt=prompt,
|
369 |
seed=seed,
|
@@ -417,9 +416,9 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
|
|
417 |
with gr.Row():
|
418 |
# Model choice radio button above additional options
|
419 |
model_choice = gr.Radio(
|
420 |
-
choices=["flux.1-krea", "flux.1-dev-realism"],
|
421 |
label="Select Model",
|
422 |
-
value="flux.1-krea"
|
423 |
)
|
424 |
|
425 |
with gr.Accordion("Additional Options", open=False):
|
@@ -464,7 +463,7 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
|
|
464 |
minimum=0.1,
|
465 |
maximum=20.0,
|
466 |
step=0.1,
|
467 |
-
value=
|
468 |
)
|
469 |
num_inference_steps = gr.Slider(
|
470 |
label="Number of inference steps",
|
|
|
12 |
from transformers import CLIPTextModel, CLIPTokenizer, T5EncoderModel, T5TokenizerFast
|
13 |
|
14 |
# Description for the app
|
15 |
+
DESCRIPTION = """## flux realism hpc/."""
|
|
|
16 |
|
17 |
# Helper functions
|
18 |
def save_image(img):
|
|
|
362 |
zip_images=zip_images,
|
363 |
progress=progress,
|
364 |
)
|
365 |
+
elif model_choice == "flux.1-krea-dev":
|
366 |
return generate_krea(
|
367 |
prompt=prompt,
|
368 |
seed=seed,
|
|
|
416 |
with gr.Row():
|
417 |
# Model choice radio button above additional options
|
418 |
model_choice = gr.Radio(
|
419 |
+
choices=["flux.1-krea-dev", "flux.1-dev-realism"],
|
420 |
label="Select Model",
|
421 |
+
value="flux.1-krea-dev"
|
422 |
)
|
423 |
|
424 |
with gr.Accordion("Additional Options", open=False):
|
|
|
463 |
minimum=0.1,
|
464 |
maximum=20.0,
|
465 |
step=0.1,
|
466 |
+
value=4.5,
|
467 |
)
|
468 |
num_inference_steps = gr.Slider(
|
469 |
label="Number of inference steps",
|