Changes

app.py

@@ -35,17 +35,14 @@ app = Flask(__name__)
 # @app.route("/", methods=["POST"])
 def generate():
 
-    # prompt = request.form['prompt']
-    # negative_prompt = request.form['Negative prompt']
-    # num_samples = request.form['No. of samples']
+    with st.form(key="Form :", clear_on_submit = True):
+        prompt = st.text_area(label = "prompt", key="pmpt")
+        negative_prompt = st.text_area(label = "Negative prompt", key="ng_pmpt")
+        num_samples = st.number_input("No. of samples", step=1)
 
-    prompt = st.text_area(label = "prompt", key="pmpt")
-    negative_prompt = st.text_area(label = "Negative prompt", key="ng_pmpt")
-    num_samples = st.number_input("No. of samples")
+        Submit = st.form_submit_button(label='Submit')
 
-    res = st.button("Generate", type="primary")
-
-    if res:
+    if Submit:
 
         guidance_scale = 7.5
         num_inference_steps = 24