Spaces:
Runtime error
Runtime error
mrbeliever
commited on
Commit
•
a2f838b
1
Parent(s):
ac832f4
Update app.py
Browse files
app.py
CHANGED
@@ -6,6 +6,7 @@ model2 = gr.load("models/dataautogpt3/ProteusV0.2")
|
|
6 |
model3 = gr.load("models/runwayml/stable-diffusion-v1-5")
|
7 |
model4 = gr.load("models/stablediffusionapi/juggernaut-xl-v5")
|
8 |
model5 = gr.load("models/stabilityai/stable-diffusion-xl-base-1.0")
|
|
|
9 |
|
10 |
# Function to switch between models and generate image
|
11 |
def generate_image(selected_model_name, text_input):
|
@@ -19,6 +20,7 @@ models = {
|
|
19 |
"runwayml sdxl": model3,
|
20 |
"juggernaut-xl-v5": model4,
|
21 |
"stable-diffusion-xl": model5
|
|
|
22 |
}
|
23 |
|
24 |
|
|
|
6 |
model3 = gr.load("models/runwayml/stable-diffusion-v1-5")
|
7 |
model4 = gr.load("models/stablediffusionapi/juggernaut-xl-v5")
|
8 |
model5 = gr.load("models/stabilityai/stable-diffusion-xl-base-1.0")
|
9 |
+
model6 = gr.load("models/Dremmar/nsfw-xl")
|
10 |
|
11 |
# Function to switch between models and generate image
|
12 |
def generate_image(selected_model_name, text_input):
|
|
|
20 |
"runwayml sdxl": model3,
|
21 |
"juggernaut-xl-v5": model4,
|
22 |
"stable-diffusion-xl": model5
|
23 |
+
"Dreammar nsfw-xl": model6
|
24 |
}
|
25 |
|
26 |
|