Update app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 from transformers import PegasusForConditionalGeneration, PegasusTokenizer
 
-st.title("Paraphrase sentences")
+st.title("Paraphrase Generator")
 
 model_name = "tuner007/pegasus_paraphrase"
 torch_device = "cpu"
@@ -36,17 +36,16 @@ def get_response(
     return tgt_text
 
 
-num_beams = 10
-num_return_sequences = st.slider("Number of paraphrases", 1, 10, 5, 1)
-context = st.text_area(label="Enter a sentence to paraphrase", max_chars=384)
+context = st.text_input(label="Enter a sentence to paraphrase", value="How do I make a deposit?")
+
+num_return_sequences = st.sidebar.slider("Number of paraphrases", 1, 20, 10, 1)
+num_beams = num_return_sequences
+temperature = st.sidebar.slider("Temperature", 0.1, 5.0, 1.5, 0.1)
+max_length = st.sidebar.slider("Max length", 10, 100, 60, 10)
 
-with st.expander("Advanced"):
-    temperature = st.slider("Temperature", 0.1, 5.0, 1.5, 0.1)
-    max_length = st.slider("Max length", 10, 100, 60, 10)
 if context:
     response = get_response(
         context, num_return_sequences, num_beams, max_length, temperature
     )
 
-    for paraphrase in response:
-        st.write(paraphrase)
+    st.write(response)