Update app.py
Browse files
app.py
CHANGED
|
@@ -24,6 +24,7 @@ import numpy as np
|
|
| 24 |
|
| 25 |
# 1. Base Model - Using reliable SDXL models (avoiding incomplete "bait" models)
|
| 26 |
BASE_MODELS = {
|
|
|
|
| 27 |
"sdxl_base": "stabilityai/stable-diffusion-xl-base-1.0", # Most reliable choice
|
| 28 |
"realistic_vision": "SG161222/RealVisXL_V4.0", # High-quality realistic model
|
| 29 |
"anime_xl": "Linaqruf/animagine-xl-3.1", # Popular anime-style SDXL
|
|
@@ -32,7 +33,7 @@ BASE_MODELS = {
|
|
| 32 |
}
|
| 33 |
|
| 34 |
# Current model selection (change this to switch models)
|
| 35 |
-
CURRENT_MODEL_KEY = "
|
| 36 |
BASE_MODEL = BASE_MODELS[CURRENT_MODEL_KEY]
|
| 37 |
|
| 38 |
# 2. Fixed LoRAs (Auto-loaded, not user-selectable) - Using actual LoRA models
|
|
@@ -564,7 +565,7 @@ def create_interface():
|
|
| 564 |
choices=list(SUPPORTED_LANGUAGES.keys()),
|
| 565 |
label="Language (Optional)",
|
| 566 |
value="en",
|
| 567 |
-
visible=
|
| 568 |
)
|
| 569 |
|
| 570 |
# m. Generate Button
|
|
|
|
| 24 |
|
| 25 |
# 1. Base Model - Using reliable SDXL models (avoiding incomplete "bait" models)
|
| 26 |
BASE_MODELS = {
|
| 27 |
+
"nsfw":"votepurchase/novaFurryXL_illustriousV7b",
|
| 28 |
"sdxl_base": "stabilityai/stable-diffusion-xl-base-1.0", # Most reliable choice
|
| 29 |
"realistic_vision": "SG161222/RealVisXL_V4.0", # High-quality realistic model
|
| 30 |
"anime_xl": "Linaqruf/animagine-xl-3.1", # Popular anime-style SDXL
|
|
|
|
| 33 |
}
|
| 34 |
|
| 35 |
# Current model selection (change this to switch models)
|
| 36 |
+
CURRENT_MODEL_KEY = "nsfw" # Changed to working model
|
| 37 |
BASE_MODEL = BASE_MODELS[CURRENT_MODEL_KEY]
|
| 38 |
|
| 39 |
# 2. Fixed LoRAs (Auto-loaded, not user-selectable) - Using actual LoRA models
|
|
|
|
| 565 |
choices=list(SUPPORTED_LANGUAGES.keys()),
|
| 566 |
label="Language (Optional)",
|
| 567 |
value="en",
|
| 568 |
+
visible=True # Hidden for now, can be enabled later
|
| 569 |
)
|
| 570 |
|
| 571 |
# m. Generate Button
|