Update app.py

app.py

@@ -30,20 +30,24 @@ st.session_state.t2m_mod = create_model(loc=the_type)
 
 prom = st.text_input("Prompt",'')
 
-c1,c2,c3 = st.columns([2,2,2])
-c5,c6 = st.columns(2)
+c1,c2,c3 = st.columns([1,1,3])
+c4,c5 = st.columns(2)
+c6 = st.columns(1)
 
 with c1:
-  bu_1 = st.text_input("Seed",'999')
+  bu_1 = st.text_input("Seed",'666')
 with c2:
   bu_2 = st.text_input("Steps",'12')
 with c3:
   bu_3 = st.text_input("Number of Images",'1')
-with c5:
+with c4:
   sl_1 = st.slider("Width",128,1024,512,8)
-with c6:
+with c5:
   sl_2 = st.slider("hight",128,1024,512,8)
 
+with c6:
+  bu_6 = st.text_input("Guidance Scale",'7.5')
+
 st.session_state.generator = torch.Generator("cpu").manual_seed(int(bu_1))
 
 create = st.button("Imagine")
@@ -56,6 +60,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,
                     generator=generator).images[0]
       st.image(IMG)
         
@@ -64,6 +69,7 @@ if create:
         
       IMGS = model(PROMS, width=int(sl_1), height=int(sl_2),
                      num_inference_steps=int(bu_2),
+                     guidance_scale = bu_6,
                      generator=generator).images
     
       st.image(IMGS)