Update app.py

app.py

@@ -115,7 +115,8 @@ def create_readme(info, downloaded_files, link_civit=False, is_author=True, fold
     print(widget_prompts)
     tags = default_tags + civit_tags
     unpacked_tags = "\n- ".join(tags)
-    content = f"""---
+    content = f"""---f"text: {widget_prompts if widget_prompts else (info['trainedWords'][0] if info['trainedWords'][0] else '')}"
+
 license: other
 tags:
 - {unpacked_tags}
@@ -123,7 +124,7 @@ 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}
+- text: {widget_prompts if widget_prompts else (info['trainedWords'][0] if 'trainedWords' in info and len(info['trainedWords']) > 0 else '')}
 ---
 
 # {info["name"]}