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 = st.columns([1,1,3])
+c1,c2,c3,c4 = st.columns([1,1,1,2])
 c5,c6 = st.columns(2)
 
 with c1:
@@ -38,7 +38,9 @@ with c1:
 with c2:
   bu_2 = st.text_input("Steps",'12')
 with c3:
-  bu_3 = st.text_input("Number of Images",'1')
+  bu_3 = st.text_input("Guidance Scale",'7.5')
+with c4:
+  bu_4 = st.text_input("Number of Images",'1')
 with c5:
   sl_1 = st.slider("Width",128,1024,512,8)
 with c6:
@@ -48,7 +50,6 @@ st.session_state.generator = torch.Generator("cpu").manual_seed(int(bu_1))
 
 create = st.button("Imagine")
 
-# bu_3 = 7.5
 
 if create:
     model = st.session_state.t2m_mod
@@ -57,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_3,
+                    guidance_scale = bu_3,
                     generator=generator).images[0]
       st.image(IMG)
         
@@ -66,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_3,
+                     guidance_scale = bu_3,
                      generator=generator).images
     
       st.image(IMGS)