|
import streamlit as st |
|
from transformers import pipeline, set_seed |
|
|
|
pipe = pipeline("text-generation", model="gpt2") |
|
|
|
st.title("Paraphrase Generator") |
|
user_word = st.text_input("Enter a line:") |
|
|
|
if st.button("Generate Paraphrase"): |
|
if user_word: |
|
|
|
paraphrase_prompt= f"Write a Paraphrase about '{user_word}'.\n" |
|
|
|
paraphrase = pipe(paraphrase_prompt, max_length=200, do_sample=True, num_return_sequences=1)[0]["generated_text"] |
|
st.markdown("**Paraphrase:**") |
|
st.markdown(paraphrase) |
|
else: |
|
st.warning("Please enter a line.") |
|
|