ArticlePara / apps /paraphraseApp.py
segestic's picture
Update apps/paraphraseApp.py
b996afb
raw
history blame contribute delete
619 Bytes
import streamlit as st
from paraphraser import get_paraphrased_sentences, model, tokenizer
def app():
st.title('Paraphraser')
st.write('Please provide the text to be paraphrased')
user_input = st.text_area('Enter text','')
paraphraseNo = st.slider('Number of Parapharases',1,10,2)
if st.button('Paraphrase'):
output = get_paraphrased_sentences(model, tokenizer, user_input, num_beams=10, num_return_sequences=paraphraseNo)
st.write("Paraphrased Text: ")
st.write(output)
##get_paraphrased_sentences(model, tokenizer, sentence, num_beams=10, num_return_sequences=10)