Update app.py

T5 task sentiment analysis

app.py

@@ -281,24 +281,44 @@
 #-----------------------------------------------------------------------------------
 # 12. Text-to-Text Generation using the T5 model - Task 3 Translation.
 
+# from transformers import T5ForConditionalGeneration, T5Tokenizer
+# import gradio as grad
+
+# text2text_tkn= T5Tokenizer.from_pretrained("t5-small")
+# mdl = T5ForConditionalGeneration.from_pretrained("t5-small")
+
+# def text2text_translation(text):
+#     # English to German
+#     # inp = "translate English to German:: "+text
+#     # English to Frendh
+#     inp = "translate English to French:: " +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="French Translation")
+
+# grad.Interface(text2text_translation, inputs=para, outputs=out).launch()
+
+#-----------------------------------------------------------------------------------
+# 13. Text-to-Text Generation using the T5 model - Task 4 sentiment analysis.
+
 from transformers import T5ForConditionalGeneration, T5Tokenizer
 import gradio as grad
 
 text2text_tkn= T5Tokenizer.from_pretrained("t5-small")
 mdl = T5ForConditionalGeneration.from_pretrained("t5-small")
 
-def text2text_translation(text):
-    # English to German
-    # inp = "translate English to German:: "+text
-    # English to Frendh
-    inp = "translate English to French:: " +text
+def text2text_sentiment(text):
+    inp = "sst2 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="German Translation")
-
-grad.Interface(text2text_translation, inputs=para, outputs=out).launch()
+out=grad.Textbox(lines=1, label="Sentiment")
 
+grad.Interface(text2text_sentiment, inputs=para, outputs=out).launch()