Update app.py

app.py

@@ -1,16 +1,18 @@
 import streamlit as st
-from transformers import pipeline, set_seed
+from transformers import pipeline
 
-pipe = pipeline("text-generation", model="google/flan-t5-base")
+# Create a text2text-generation pipeline with the "google/flan-t5-base" model
+paraphrase_generator = pipeline("text2text-generation", model="google/flan-t5-base")
 
 st.title("Paraphrase Generator")
-user_word = st.text_input("Enter a line:")
+
+user_input = st.text_input("Enter a line:")
 
 if st.button("Generate Paraphrase"):
-    if user_word:
-        paraphrase_prompt= f"Write a Paraphrase on '{user_word}'.\n"
-        set_seed(42)
-        paraphrase = pipe(paraphrase_prompt, max_length=100, do_sample=True, num_return_sequences=1)[0]["generated_text"]
+    if user_input:
+        # Use a prompt to control the paraphrasing
+        paraphrase_prompt = f"Paraphrase the following line: '{user_input}'"
+        paraphrase = paraphrase_generator(paraphrase_prompt, max_length=100, do_sample=True)[0]["generated_text"]
         st.markdown("**Paraphrase:**")
         st.markdown(paraphrase)
     else: