Update app.py

app.py

@@ -109,11 +109,19 @@ def create_readme(info, downloaded_files, link_civit=False, is_author=True, fold
     original_url = f"https://civitai.com/models/{info['modelId']}"
     link_civit_disclaimer = f'([CivitAI]({original_url}))'
     non_author_disclaimer = f'This model was originally uploaded on [CivitAI]({original_url}), by [{info["creator"]}](https://civitai.com/user/{info["creator"]}/models). The information below was provided by the author on CivitAI:'
-    default_tags = ["text-to-image", "stable-diffusion", "lora", "diffusers"]
+    default_tags = ["text-to-image", "stable-diffusion", "lora", "diffusers", "template:sd-lora"]
     civit_tags = [t for t in info["tags"] if t not in default_tags]
-    widget_prompts = "\n- text: ".join(['"' + prompt.replace('"', '\\"') + '"' for prompt in downloaded_files["imagePrompt"] if prompt])
     tags = default_tags + civit_tags
     unpacked_tags = "\n- ".join(tags)
+
+    widget_content = ""
+    for index, (prompt, image) in enumerate(zip(downloaded_files["imagePrompt"], downloaded_files["imageName"])):
+        widget_content += f"""- text: '{prompt}'
+  output:
+    url: >-
+      {image}
+"""
+
     content = f"""---
 license: other
 tags:
@@ -122,11 +130,13 @@ tags:
 base_model: stabilityai/stable-diffusion-xl-base-1.0
 instance_prompt: {info['trainedWords'][0] if 'trainedWords' in info and len(info['trainedWords']) > 0 else ''}
 widget:
-- text: {widget_prompts if widget_prompts else (info['trainedWords'][0] if 'trainedWords' in info and len(info['trainedWords']) > 0 else '')}
+{widget_content}
 ---
 
 # {info["name"]} 
 
+<Gallery />
+
 {non_author_disclaimer if not is_author else ''}
 
 ![Image 0]({downloaded_files["imageName"][0]})