Update app.py

app.py

@@ -42,9 +42,9 @@ with c5:
 
 create = st.button("Imagine")
 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]
+    # 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)
 
     if int(bu_3) == 1 :
@@ -58,9 +58,9 @@ if create:
       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),
+      IMGS = model(PROMS, width=int(sl_1), height=int(sl_2),
                      num_inference_steps=int(bu_2),
                      generator=generator).images
     
-      IMGS = np.hstack(IMGS)
+      # IMGS = np.hstack(IMGS)
       st.image(IMGS)