Update app.py

app.py

@@ -144,12 +144,12 @@ widget:
 {info["description"]}
 
 """
-    #for index, (image, prompt) in enumerate(zip(downloaded_files["imageName"], downloaded_files["imagePrompt"])):
-    #    if index == 1:
-    #        content += f"## Image examples for the model:\n![Image {index}]({image})\n> {prompt}\n"
-    #    elif index > 1:
-    #        content += f"\n![Image {index}]({image})\n> {prompt}\n"
-    #readme_content += content + "\n"
+    for index, (image, prompt) in enumerate(zip(downloaded_files["imageName"], downloaded_files["imagePrompt"])):
+        if index == 1:
+            content += f"## Image examples for the model:\n![Image {index}]({image})\n> {prompt}\n"
+        elif index > 1:
+            content += f"\n![Image {index}]({image})\n> {prompt}\n"
+    readme_content += content + "\n"
     print(readme_content)
     with open(f"{folder}/README.md", "w") as file:
         file.write(readme_content)