Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -2,7 +2,9 @@ import streamlit as st
|
|
2 |
from transformers import PegasusForConditionalGeneration, PegasusTokenizer
|
3 |
|
4 |
st.title("Paraphrase with Pegasus")
|
5 |
-
|
|
|
|
|
6 |
model_name = "tuner007/pegasus_paraphrase"
|
7 |
torch_device = "cpu"
|
8 |
tokenizer = PegasusTokenizer.from_pretrained(model_name)
|
@@ -25,7 +27,7 @@ def get_response(
|
|
25 |
|
26 |
num_beams = 10
|
27 |
num_return_sequences = st.slider("Number of paraphrases", 1, 10, 5, 1)
|
28 |
-
context = st.text_area(label="
|
29 |
|
30 |
with st.expander("Advanced"):
|
31 |
temperature = st.slider("Temperature", 0.1, 5.0, 1.5, 0.1)
|
|
|
2 |
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"
|
10 |
tokenizer = PegasusTokenizer.from_pretrained(model_name)
|
|
|
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"):
|
33 |
temperature = st.slider("Temperature", 0.1, 5.0, 1.5, 0.1)
|