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):
|