Spaces:

bixentemal
/

q_n_a_roberta

Runtime error

bixentemal commited on Mar 15, 2023

Commit

bebff77

•

1 Parent(s): 22151c8

Gramatical acceptance T5

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ import gradio as grad
 text2text_tkn= T5Tokenizer.from_pretrained("t5-small")
 mdl = T5ForConditionalGeneration.from_pretrained("t5-small")
 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)

 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")
     tokens = mdl.generate(**enc)
     response=text2text_tkn.batch_decode(tokens)