Update app.py

app.py

@@ -66,6 +66,16 @@ def text2text_acceptable_sentence(text):
 
     return response
 
+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
+
 # question
 # context = grad.Textbox(lines = 10, label = 'English', placeholder = 'Context')
 # ans = grad.Textbox(lines = 1, label = 'Answer')
@@ -80,12 +90,17 @@ def text2text_acceptable_sentence(text):
 # out = grad.Textbox(lines = 1, label = 'German Translation')
 
 # sentiment
-para = grad.Textbox(lines = 1, label = 'English Text', placeholder = 'Text in English')
-out = grad.Textbox(lines = 1, label = 'Sentiment')
+# para = grad.Textbox(lines = 1, label = 'English Text', placeholder = 'Text in English')
+# out = grad.Textbox(lines = 1, label = 'Sentiment')
+
+# # grammatical acceptance
+# 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')
 
-# grammatical acceptance
-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')
+# paraphrase
+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 = 'Paraphrase')
 
 grad.Interface(
    # text2text,
@@ -93,7 +108,9 @@ grad.Interface(
     # text2text_summary,
     # text2text_translate,
     # text2text_sentiment,
-    text2text_acceptable_sentence,
-    inputs = para,
+    # text2text_acceptable_sentence,
+    text2text_paraphrase,
+    # inputs = para,
+    inputs = [sent1, sent2],
     outputs = out
 ).launch()