Update app.py

app.py

@@ -134,7 +134,14 @@ def create_readme(info, downloaded_files, user_repo_id, link_civit=False, is_aut
     
     trained_words = info['trainedWords'] if 'trainedWords' in info and info['trainedWords'] else []
     formatted_words = ', '.join(f'`{word}`' for word in trained_words)
-
+    if formatted_words:
+        trigger_words_section = f"""## Trigger words
+    
+    You should use {formatted_words} to trigger the image generation.
+    """
+    else:
+        trigger_words_section = ""
+    
     widget_content = ""
     for index, (prompt, image) in enumerate(zip(downloaded_files["imagePrompt"], downloaded_files["imageName"])):
         escaped_prompt = prompt.replace("'", "''")
@@ -169,9 +176,7 @@ widget:
 
 {info["description"]}
 
-## Trigger words
-
-You should use {formatted_words} to trigger the image generation.
+{trigger_words_section}
 
 ## Download model
 
@@ -186,7 +191,7 @@ from diffusers import AutoPipelineForText2Image
 import torch
 
 pipeline = AutoPipelineForText2Image.from_pretrained('{info["baseModel"]}', torch_dtype=torch.float16).to('cuda')
-pipeline.load_lora_weights('{user_repo_id}', weight_name='{downloaded_files["weightName"]}')
+pipeline.load_lora_weights('{user_repo_id}', weight_name='{downloaded_files["weightName"][0]}')
 image = pipeline('{prompt if prompt else (formatted_words if formatted_words else 'Your custom prompt')}').images[0]
 ```