multimodalart HF Staff commited on
Commit
ff20abd
·
verified ·
1 Parent(s): 2c21d02

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
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} output:
 
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):