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,c6 = st.columns([1,1,1,2])
+c1,c2,c3,c6 = st.columns([1,1,1,1])
 c4,c5 = st.columns(2)
 
 with c1:
@@ -58,7 +58,7 @@ 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_6,
+                    guidance_scale = float(bu_6),
                     generator=generator).images[0]
       st.image(IMG)
         
@@ -67,7 +67,7 @@ if create:
         
       IMGS = model(PROMS, width=int(sl_1), height=int(sl_2),
                      num_inference_steps=int(bu_2),
-                     guidance_scale = bu_6,
+                     guidance_scale = float(bu_6),
                      generator=generator).images
     
       st.image(IMGS)