Update app.py

app.py

@@ -45,4 +45,22 @@ if create:
     generator = torch.Generator("cpu").manual_seed(int(bu_1))
     model = st.session_state.t2m_mod
     img = model(prom, width=int(sl_1), height=int(sl_2), num_inference_steps=int(bu_2), generator=generator).images[0]
-    st.image(img)
+    # st.image(img)
+
+    if int(bu_3) == 1 :
+      generator = torch.Generator("cpu").manual_seed(int(bu_1))
+      model = st.session_state.t2m_mod
+      IMG = model(prom, width=int(sl_1), height=int(sl_2),
+                    num_inference_steps=int(bu_2),
+                    generator=generator).images[0]
+      st.image(IMG)
+    else :
+      generator = torch.Generator("cpu").manual_seed(int(bu_1))
+      PROMS = [prom]*int(bu_3)
+      model = st.session_state.t2m_mod
+      IMGS = model(prom, width=int(sl_1), height=int(sl_2),
+                     num_inference_steps=int(bu_2),
+                     generator=generator).images
+    
+      IMGS = np.hstack(IMGS)
+      st.image(IMGS)