Update app.py

app.py

@@ -30,7 +30,16 @@ st.session_state.t2m_mod = create_model(loc=the_type)
 
 prom = st.text_input("Prompt",'')
 
+neg_prom = st.text_input("Negative Prompt",'')
+
+style = st.selectbox("Image style",("Cyberpunk",
+                                    "Picasso",
+                                   "Real-world specific",
+                                   "Digital Art",
+                                   "Aesthetics"))
+
 c1,c2,c3,c6 = st.columns([1,1,1,1])
+c8 = st.columns([1,1,1,1])
 c4,c5 = st.columns(2)
 
 with c1:
@@ -56,7 +65,7 @@ if create:
     generator = st.session_state.generator
 
     if int(bu_3) == 1 :
-      IMG = model(prom, width=int(sl_1), height=int(sl_2),
+      IMG = model(prom, negative_prompt = neg_prom, width=int(sl_1), height=int(sl_2),
                     num_inference_steps=int(bu_2),
                     guidance_scale = float(bu_6),
                     generator=generator).images[0]
@@ -65,7 +74,7 @@ if create:
     else :
       PROMS = [prom]*int(bu_3)
         
-      IMGS = model(PROMS, width=int(sl_1), height=int(sl_2),
+      IMGS = model(PROMS, negative_prompt = neg_prom, width=int(sl_1), height=int(sl_2),
                      num_inference_steps=int(bu_2),
                      guidance_scale = float(bu_6),
                      generator=generator).images