Update app.py

app.py

@@ -28,23 +28,30 @@ st.write("Enter seed.")
 seed_input = st.text_area("Set seed")
 
 
-
 st.write("Enter max length.")
 maxLength = st.text_area("max length")
 
 # Convert seed input to integer
 try:
     seed = int(seed_input)
-    maxLength = int(maxLength)
+    max_length = int(maxLength)
 except ValueError:
     seed = None
-    maxLength = None
+    max_length = None
     
 set_seed(seed)
 
 
+if st.button("Generate Text"):
+    # Use default values or handle None appropriately
+    if seed is not None:
+        set_seed(seed)
+
+    if text:
+        # Generate text
+        out = generator(text, max_length=max_length, num_return_sequences=1)
+        st.json(out)
+        st.write(f"Reply: {out[0]['generated_text']}")
+
+
 
-if text and seed and maxLength:
-    out = generator(text, max_length=maxLength, num_return_sequences=5)
-    st.json(out)
-    st.write(f"Reply: {out[0]['generated_text']}")