Spaces:
Runtime error
Runtime error
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 |
|