Commit
·
2f7f340
1
Parent(s):
6518260
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,31 +8,17 @@ model_id = int(os.getenv("Model"))
|
|
| 8 |
|
| 9 |
#stable-diffusion-xl-base-1.0 0 - base model
|
| 10 |
#Colossus_Project_XL 1 - better people
|
| 11 |
-
#
|
| 12 |
#JuggernautXL_version5 3 - better faces
|
| 13 |
#JuggernautXL_version6 4 - better faces
|
| 14 |
#RealVisXL_V2.0 5 - realistic
|
| 15 |
-
#AlbedoBaseXL_v11 6 - realistic
|
| 16 |
-
#BetterThanWords_v20_sdxl 7 - nsfw
|
| 17 |
-
#AcornIsSpinning_acornXLV1 8 - nsfw
|
| 18 |
-
#PyrosNSFWSDXL_v04. 9 - nsfw
|
| 19 |
-
#AltXL_v60 10 - realistic
|
| 20 |
-
#SDXXXL_v10 11 - nsfw
|
| 21 |
-
#DAC_PhotoEcstacy_XL_SE_V2 12 - doll people
|
| 22 |
|
| 23 |
model_url_list = ["stabilityai/stable-diffusion-xl-base-1.0/blob/main/sd_xl_base_1.0.safetensors",
|
| 24 |
"Krebzonide/Colossus_Project_XL/blob/main/colossusProjectXLSFW_v202BakedVAE.safetensors",
|
| 25 |
-
"Krebzonide/
|
| 26 |
"Krebzonide/JuggernautXL_version5/blob/main/juggernautXL_version5.safetensors",
|
| 27 |
"Krebzonide/JuggernautXL_version5/blob/main/juggernautXL_version6Rundiffusion.safetensors",
|
| 28 |
-
"SG161222/RealVisXL_V2.0/blob/main/RealVisXL_V2.0.safetensors"
|
| 29 |
-
"Krebzonide/AlbedoBaseXL_v11/blob/main/albedobaseXL_v11.safetensors",
|
| 30 |
-
"Krebzonide/BetterThanWords_v20_sdxl/blob/main/betterThanWords_v20.safetensors",
|
| 31 |
-
"Krebzonide/AcornIsSpinning_acornXLV1/blob/main/acornIsSpinning_acornxlV1.safetensors",
|
| 32 |
-
"Krebzonide/PyrosNSFWSDXL_v04/blob/main/pyrosNSFWSDXL_v04.safetensors",
|
| 33 |
-
"Krebzonide/AltXL_v60/blob/main/altxl_v60.safetensors",
|
| 34 |
-
"Krebzonide/SDXXXL_v10/blob/main/sdxxxl_v10.safetensors",
|
| 35 |
-
"Krebzonide/DAC_PhotoEcstacy_XL_SE_V2/blob/main/dacPhotoecstasyXLSE_v20.safetensors"]
|
| 36 |
|
| 37 |
css = """
|
| 38 |
.btn-green {
|
|
@@ -104,19 +90,18 @@ with gr.Blocks(css=css) as demo:
|
|
| 104 |
negative_prompt = gr.Textbox(label="Negative Prompt")
|
| 105 |
submit_btn = gr.Button("Generate", elem_classes="btn-green")
|
| 106 |
with gr.Row():
|
| 107 |
-
samp_steps = gr.Slider(1,
|
| 108 |
-
cfg_scale = gr.Slider(1, 10, value=
|
| 109 |
-
batch_size = gr.Slider(1,
|
| 110 |
with gr.Row():
|
| 111 |
height = gr.Slider(label="Height", value=1024, minimum=8, maximum=1536, step=8)
|
| 112 |
-
width = gr.Slider(label="Width", value=1024, minimum=8, maximum=
|
| 113 |
with gr.Row():
|
| 114 |
pixels = gr.Number(label="Pixel Ratio", value=1, interactive=False)
|
| 115 |
seed = gr.Number(label="Seed", value=-1, minimum=-1, precision=0)
|
| 116 |
gallery = gr.Gallery(label="Generated images")
|
| 117 |
with gr.Row():
|
| 118 |
lastSeed = gr.Number(label="Last Seed", value=-1, interactive=False)
|
| 119 |
-
|
| 120 |
ex = gr.Examples(examples=examples, inputs=[prompt, negative_prompt])
|
| 121 |
submit_btn.click(generate, [prompt, negative_prompt, samp_steps, cfg_scale, batch_size, seed, height, width], [gallery, lastSeed], queue=True)
|
| 122 |
height.release(update_pixel_ratio, [height, width], [pixels, height], queue=False)
|
|
|
|
| 8 |
|
| 9 |
#stable-diffusion-xl-base-1.0 0 - base model
|
| 10 |
#Colossus_Project_XL 1 - better people
|
| 11 |
+
#AlbedoBaseXL_v11 2 - realistic
|
| 12 |
#JuggernautXL_version5 3 - better faces
|
| 13 |
#JuggernautXL_version6 4 - better faces
|
| 14 |
#RealVisXL_V2.0 5 - realistic
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 15 |
|
| 16 |
model_url_list = ["stabilityai/stable-diffusion-xl-base-1.0/blob/main/sd_xl_base_1.0.safetensors",
|
| 17 |
"Krebzonide/Colossus_Project_XL/blob/main/colossusProjectXLSFW_v202BakedVAE.safetensors",
|
| 18 |
+
"Krebzonide/AlbedoBaseXL_v11/blob/main/albedobaseXL_v11.safetensors",
|
| 19 |
"Krebzonide/JuggernautXL_version5/blob/main/juggernautXL_version5.safetensors",
|
| 20 |
"Krebzonide/JuggernautXL_version5/blob/main/juggernautXL_version6Rundiffusion.safetensors",
|
| 21 |
+
"SG161222/RealVisXL_V2.0/blob/main/RealVisXL_V2.0.safetensors"]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 22 |
|
| 23 |
css = """
|
| 24 |
.btn-green {
|
|
|
|
| 90 |
negative_prompt = gr.Textbox(label="Negative Prompt")
|
| 91 |
submit_btn = gr.Button("Generate", elem_classes="btn-green")
|
| 92 |
with gr.Row():
|
| 93 |
+
samp_steps = gr.Slider(1, 30, value=20, step=1, label="Sampling steps")
|
| 94 |
+
cfg_scale = gr.Slider(1, 10, value=4, step=0.5, label="Guidance scale")
|
| 95 |
+
batch_size = gr.Slider(1, 1, value=1, step=1, label="Batch size")
|
| 96 |
with gr.Row():
|
| 97 |
height = gr.Slider(label="Height", value=1024, minimum=8, maximum=1536, step=8)
|
| 98 |
+
width = gr.Slider(label="Width", value=1024, minimum=8, maximum=1024, step=8)
|
| 99 |
with gr.Row():
|
| 100 |
pixels = gr.Number(label="Pixel Ratio", value=1, interactive=False)
|
| 101 |
seed = gr.Number(label="Seed", value=-1, minimum=-1, precision=0)
|
| 102 |
gallery = gr.Gallery(label="Generated images")
|
| 103 |
with gr.Row():
|
| 104 |
lastSeed = gr.Number(label="Last Seed", value=-1, interactive=False)
|
|
|
|
| 105 |
ex = gr.Examples(examples=examples, inputs=[prompt, negative_prompt])
|
| 106 |
submit_btn.click(generate, [prompt, negative_prompt, samp_steps, cfg_scale, batch_size, seed, height, width], [gallery, lastSeed], queue=True)
|
| 107 |
height.release(update_pixel_ratio, [height, width], [pixels, height], queue=False)
|