Update app.py

app.py

@@ -80,4 +80,4 @@ if submitted :
 
     st.write("Done!")
 
-    st.image(images, width=150, caption=f"Generated Images with <{prompt}>")
+    st.image(images, width=300, caption=[f"Generated Images with <{prompt}>" for i in range(len(images))])