Wootang01 commited on
Commit
f871f8b
1 Parent(s): 4bda66e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,7 +3,7 @@ from transformers import PegasusForConditionalGeneration, PegasusTokenizer
3
 
4
  st.title("Paraphrase with Pegasus")
5
  st.write("Paraphrase means to express meaning using different words. Pegasus refers to a natural language processing model.")
6
- st.write("Write or paste text below, select a number of paraphrases and enter. The machine will attempt to produce your selected number of paraphrases.")
7
 
8
  model_name = "tuner007/pegasus_paraphrase"
9
  torch_device = "cpu"
@@ -26,7 +26,7 @@ def get_response(
26
  return tgt_text
27
 
28
  num_beams = 10
29
- num_return_sequences = st.slider("Number of paraphrases", 1, 10, 5, 1)
30
  context = st.text_area(label="Write or paste text", max_chars=384)
31
 
32
  with st.expander("Advanced"):
 
3
 
4
  st.title("Paraphrase with Pegasus")
5
  st.write("Paraphrase means to express meaning using different words. Pegasus refers to a natural language processing model.")
6
+ st.write("Write or paste text below, select a number of paraphrases and enter. The machine will attempt to produce your selected number of paraphrases. You can also select advanced features.")
7
 
8
  model_name = "tuner007/pegasus_paraphrase"
9
  torch_device = "cpu"
 
26
  return tgt_text
27
 
28
  num_beams = 10
29
+ num_return_sequences = st.slider("Number of paraphrases", 1, 5, 3, 1)
30
  context = st.text_area(label="Write or paste text", max_chars=384)
31
 
32
  with st.expander("Advanced"):