Update app.py

app.py

@@ -39,7 +39,7 @@ def paraphraze(text, how_many=1):
     paraphrase.append(a)
     paraphrase2 = [' '.join(x) for x in paraphrase]
     paraphrase3 = [' '.join(x for x in paraphrase2) ]
-  paraphrased_text = str(paraphrase3).strip('[]').strip("'")
+  paraphrased_text = paraphrase #str(paraphrase3).strip('[]').strip("'")
   return paraphrased_text
 
 
@@ -49,23 +49,23 @@ def summarize(text):
   return paraphrased_text
 ########################################################################################################
 
-# from transformers import *
+from transformers import *
 
-# model = PegasusForConditionalGeneration.from_pretrained("tuner007/pegasus_paraphrase")
-# tokenizer = PegasusTokenizerFast.from_pretrained("tuner007/pegasus_paraphrase")
+model = PegasusForConditionalGeneration.from_pretrained("tuner007/pegasus_paraphrase")
+tokenizer = PegasusTokenizerFast.from_pretrained("tuner007/pegasus_paraphrase")
 
 
-# def get_paraphrased_sentences(model, tokenizer, sentence, num_return_sequences=5, num_beams=5):
-#   # tokenize the text to be form of a list of token IDs
-#   inputs = tokenizer([sentence], truncation=False, padding="longest", return_tensors="pt")
-#   # generate the paraphrased sentences
-#   outputs = model.generate(
-#     **inputs,
-#     num_beams=num_beams,
-#     num_return_sequences=num_return_sequences,
-#   )
-#   # decode the generated sentences using the tokenizer to get them back to text
-#   return tokenizer.batch_decode(outputs, skip_special_tokens=True)
+def get_paraphrased_sentences(model, tokenizer, sentence, num_return_sequences=5, num_beams=5):
+  # tokenize the text to be form of a list of token IDs
+  inputs = tokenizer([sentence], truncation=False, padding="longest", return_tensors="pt")
+  # generate the paraphrased sentences
+  outputs = model.generate(
+    **inputs,
+    num_beams=num_beams,
+    num_return_sequences=num_return_sequences,
+  )
+  # decode the generated sentences using the tokenizer to get them back to text
+  return tokenizer.batch_decode(outputs, skip_special_tokens=True)
 
 
 
@@ -81,11 +81,16 @@ def app():
         user_input = st.text_area('Enter text','', height=200)
         
         paraphraseNo = st.slider('Number of Parapharases',1,10,2)
-        if st.button('Paraphrase'):
+        if st.button('Single-Paraphrase'):
             with st.spinner(text="This may take a moment..."):
-                #output = get_paraphrased_sentences(model, tokenizer, user_input, num_beams=10, num_return_sequences=paraphraseNo)    
-                output = summarize(user_input)    
-    
+                output = summarize(user_input) 
+                
+        if st.button('Multiple-Paraphrase'):
+            with st.spinner(text="This may take a moment..."):
+                output = get_paraphrased_sentences(model, tokenizer, user_input, num_beams=10, num_return_sequences=paraphraseNo)    
+                
+
+                
     #with spacer:
     
     with col2: