Update app.py

app.py

@@ -58,6 +58,14 @@ def text2text_sentiment(text):
 
     return response
 
+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
+
 # question
 # context = grad.Textbox(lines = 10, label = 'English', placeholder = 'Context')
 # ans = grad.Textbox(lines = 1, label = 'Answer')
@@ -75,12 +83,17 @@ def text2text_sentiment(text):
 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')
+
 grad.Interface(
    # text2text,
     # inputs = [context, ans],
     # text2text_summary,
     # text2text_translate,
-    text2text_sentiment,
+    # text2text_sentiment,
+    text2text_acceptable_sentence,
     inputs = para,
     outputs = out
 ).launch()