Krebzonide
commited on
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)
|