paraphrasing task T5

app.py

@@ -2,12 +2,15 @@ from transformers import T5ForConditionalGeneration, T5Tokenizer
 import gradio as grad
 text2text_tkn= T5Tokenizer.from_pretrained("t5-small")
 mdl = T5ForConditionalGeneration.from_pretrained("t5-small")
-def text2text_sentiment(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")
-out=grad.Textbox(lines=1, label="Sentiment")
-grad.Interface(text2text_sentiment, inputs=para, outputs=out).launch()
+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_paraphrase, inputs=[sent1,sent2], outputs=out).launch()