Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -16,7 +16,7 @@ def generate_images(text, selected_model):
|
|
| 16 |
model = model3
|
| 17 |
elif selected_model == "Model 4 (Flux NSFW)":
|
| 18 |
model = model4
|
| 19 |
-
elif selected_model == "Model
|
| 20 |
model = model5
|
| 21 |
else:
|
| 22 |
return "Invalid model selection."
|
|
|
|
| 16 |
model = model3
|
| 17 |
elif selected_model == "Model 4 (Flux NSFW)":
|
| 18 |
model = model4
|
| 19 |
+
elif selected_model == "Model 5 (Lora Uncensored)":
|
| 20 |
model = model5
|
| 21 |
else:
|
| 22 |
return "Invalid model selection."
|