Update app.py

T5 task sentence paraphrasing

app.py

@@ -324,7 +324,28 @@
 # grad.Interface(text2text_sentiment, inputs=para, outputs=out).launch()
 
 #-----------------------------------------------------------------------------------
-# 14. Text-to-Text Generation using the T5 model - Task 5 grammar check.
+# 14. Text-to-Text Generation using the T5 model - Task 5 grammar check - this doesn't work great unfortunately.
+
+# from transformers import T5ForConditionalGeneration, T5Tokenizer
+# import gradio as grad
+
+# text2text_tkn= T5Tokenizer.from_pretrained("t5-small")
+# mdl = T5ForConditionalGeneration.from_pretrained("t5-small")
+
+# def text2text_acceptable_sentence(text):
+#     inp = "cola sentence: "+text
+#     enc = text2text_tkn(inp, return_tensors="pt")
+#     tokens = mdl.generate(**enc)
+#     response=text2text_tkn.batch_decode(tokens)
+#     return response
+
+# para=grad.Textbox(lines=1, label="English Text", placeholder="Text in English")
+# out=grad.Textbox(lines=1, label="Whether the sentence is acceptable or not")
+
+# grad.Interface(text2text_acceptable_sentence, inputs=para, outputs=out).launch()
+
+#-----------------------------------------------------------------------------------
+# 15. Text-to-Text Generation using the T5 model - Task 6 sentence paraphasing
 
 from transformers import T5ForConditionalGeneration, T5Tokenizer
 import gradio as grad
@@ -332,14 +353,17 @@ import gradio as grad
 text2text_tkn= T5Tokenizer.from_pretrained("t5-small")
 mdl = T5ForConditionalGeneration.from_pretrained("t5-small")
 
-def text2text_acceptable_sentence(text):
-    inp = "cola sentence: "+text
-    enc = text2text_tkn(inp, return_tensors="pt")
+def text2text_paraphrase(sentence1,sentence2):
+    inp1 = "mrpc sentence1: "+sentence1
+    inp2 = "sentence2: "+sentence2
+    combined_inp=inp1+" "+inp2
+    enc = text2text_tkn(combined_inp, return_tensors="pt")
     tokens = mdl.generate(**enc)
     response=text2text_tkn.batch_decode(tokens)
     return response
 
-para=grad.Textbox(lines=1, label="English Text", placeholder="Text in English")
+sent1=grad.Textbox(lines=1, label="Sentence1", placeholder="Text in English")
+sent2=grad.Textbox(lines=1, label="Sentence2", placeholder="Text in English")
 out=grad.Textbox(lines=1, label="Whether the sentence is acceptable or not")
 
-grad.Interface(text2text_acceptable_sentence, inputs=para, outputs=out).launch()
+grad.Interface(text2text_paraphrase, inputs=[sent1,sent2], outputs=out).launch()