Update app.py

app.py

@@ -30,7 +30,7 @@ st.session_state.t2m_mod = create_model(loc=the_type)
 
 prom = st.text_input("Prompt",'')
 
-c1,c2,c3,c4 = st.columns([1,1,1,2])
+c1,c2,c3 = st.columns([2,2,2])
 c5,c6 = st.columns(2)
 
 with c1:
@@ -56,7 +56,6 @@ if create:
     if int(bu_3) == 1 :
       IMG = model(prom, width=int(sl_1), height=int(sl_2),
                     num_inference_steps=int(bu_2),
-                    # guidance_scale = bu_3,
                     generator=generator).images[0]
       st.image(IMG)
         
@@ -65,7 +64,6 @@ if create:
         
       IMGS = model(PROMS, width=int(sl_1), height=int(sl_2),
                      num_inference_steps=int(bu_2),
-                     # guidance_scale = bu_3,
                      generator=generator).images
     
       st.image(IMGS)