Update app.py
Browse files
app.py
CHANGED
|
@@ -247,8 +247,8 @@ MAX_IMAGE_SIZE = 2048
|
|
| 247 |
|
| 248 |
# Define available models
|
| 249 |
AVAILABLE_MODELS = {
|
| 250 |
-
"
|
| 251 |
-
"
|
| 252 |
# Add more models here as needed
|
| 253 |
}
|
| 254 |
|
|
@@ -362,7 +362,7 @@ def infer(
|
|
| 362 |
"num_inference_steps": num_inference_steps,
|
| 363 |
"model": model_name,
|
| 364 |
"use_resolution_binning": use_resolution_binning,
|
| 365 |
-
"PreUrl": "https://huggingface.co/spaces/Menyu/
|
| 366 |
}
|
| 367 |
|
| 368 |
# Add metadata to the image
|
|
|
|
| 247 |
|
| 248 |
# Define available models
|
| 249 |
AVAILABLE_MODELS = {
|
| 250 |
+
"NoobAI XL 1.1": "noobaiXLNAIXL_epsilonPred11Version.safetensors",
|
| 251 |
+
"naixl Mmmmix v5": "naixlMmmmix_v50.safetensors",
|
| 252 |
# Add more models here as needed
|
| 253 |
}
|
| 254 |
|
|
|
|
| 362 |
"num_inference_steps": num_inference_steps,
|
| 363 |
"model": model_name,
|
| 364 |
"use_resolution_binning": use_resolution_binning,
|
| 365 |
+
"PreUrl": "https://huggingface.co/spaces/Menyu/MenYuImg"
|
| 366 |
}
|
| 367 |
|
| 368 |
# Add metadata to the image
|