Update app.py

app.py

@@ -6,13 +6,16 @@ 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="English Text", placeholder="Text in English")
+sent2=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()
+grad.Interface(text2text_paraphrase, inputs=[sent1,sent2], outputs=out).launch()