Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -29,13 +29,10 @@ MODEL_MAPPING_IMAGE = {
|
|
| 29 |
"SD 3.5 Large Turbo": "stabilityai/stable-diffusion-3.5-large-turbo",
|
| 30 |
"Flux.1 D": "black-forest-labs/FLUX.1-dev",
|
| 31 |
"Flux.1 S": "black-forest-labs/FLUX.1-schnell",
|
| 32 |
-
"Pony": " ગુરુવર્ય/pony-diffusion-v6-xl", # Example, ensure this is a valid HF repo or use official if available
|
| 33 |
-
"Illustrious": " ગુરુવર્ય/illustrious_ai_art_generator_v1", # Example, ensure this is a valid HF repo
|
| 34 |
}
|
| 35 |
|
| 36 |
MODEL_MAPPING_VIDEO = {
|
| 37 |
"LTXV": "Lightricks/LTX-Video-0.9.7-dev",
|
| 38 |
-
# Hunyuan Video is handled specially based on user choice
|
| 39 |
"Wan Video 1.3B t2v": "Wan-AI/Wan2.1-T2V-1.3B-Diffusers",
|
| 40 |
"Wan Video 14B t2v": "Wan-AI/Wan2.1-T2V-14B-Diffusers",
|
| 41 |
"Wan Video 14B i2v 480p": "Wan-AI/Wan2.1-I2V-14B-480P-Diffusers",
|
|
@@ -360,7 +357,8 @@ def create_readme(info: Dict[str, Any], downloaded_files: Dict[str, Any], user_r
|
|
| 360 |
|
| 361 |
negative_prompt_content = f" negative_prompt: {negative_prompt}\n" if negative_prompt else ""
|
| 362 |
widget_content += f"""- text: '{escaped_prompt}'
|
| 363 |
-
{negative_prompt_content}
|
|
|
|
| 364 |
url: >-
|
| 365 |
{base_media_filename}
|
| 366 |
"""
|
|
@@ -496,6 +494,7 @@ For more details, including weighting, merging and fusing LoRAs, check the [docu
|
|
| 496 |
with open(readme_path, "w", encoding="utf-8") as file: # Added encoding
|
| 497 |
file.write(readme_content)
|
| 498 |
print(f"README.md created at {readme_path}")
|
|
|
|
| 499 |
|
| 500 |
|
| 501 |
def get_creator(username):
|
|
|
|
| 29 |
"SD 3.5 Large Turbo": "stabilityai/stable-diffusion-3.5-large-turbo",
|
| 30 |
"Flux.1 D": "black-forest-labs/FLUX.1-dev",
|
| 31 |
"Flux.1 S": "black-forest-labs/FLUX.1-schnell",
|
|
|
|
|
|
|
| 32 |
}
|
| 33 |
|
| 34 |
MODEL_MAPPING_VIDEO = {
|
| 35 |
"LTXV": "Lightricks/LTX-Video-0.9.7-dev",
|
|
|
|
| 36 |
"Wan Video 1.3B t2v": "Wan-AI/Wan2.1-T2V-1.3B-Diffusers",
|
| 37 |
"Wan Video 14B t2v": "Wan-AI/Wan2.1-T2V-14B-Diffusers",
|
| 38 |
"Wan Video 14B i2v 480p": "Wan-AI/Wan2.1-I2V-14B-480P-Diffusers",
|
|
|
|
| 357 |
|
| 358 |
negative_prompt_content = f" negative_prompt: {negative_prompt}\n" if negative_prompt else ""
|
| 359 |
widget_content += f"""- text: '{escaped_prompt}'
|
| 360 |
+
{negative_prompt_content}
|
| 361 |
+
output:
|
| 362 |
url: >-
|
| 363 |
{base_media_filename}
|
| 364 |
"""
|
|
|
|
| 494 |
with open(readme_path, "w", encoding="utf-8") as file: # Added encoding
|
| 495 |
file.write(readme_content)
|
| 496 |
print(f"README.md created at {readme_path}")
|
| 497 |
+
print(f"README.md content {readme_content}")
|
| 498 |
|
| 499 |
|
| 500 |
def get_creator(username):
|