Spaces:

SeyedAli
/

Persian-Text-Sentiment

Runtime error

SeyedAli commited on 6 days ago

Commit

1f45a5d

•

1 Parent(s): 9f2e705

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ import torch
 from transformers import AutoModelForSequenceClassification,AutoTokenizer,pipeline
 model = AutoModelForSequenceClassification.from_pretrained('SeyedAli/Persian-Text-Sentiment-Bert-V1')
-tokenizer = AutoTokenizer.from_pretrained('SeyedAli/Persian-Text-Sentiment-Bert-V1')
 def Sentiment(text):
     pipline = pipeline(task="text-classification", model=model, tokenizer=tokenizer)

 from transformers import AutoModelForSequenceClassification,AutoTokenizer,pipeline
 model = AutoModelForSequenceClassification.from_pretrained('SeyedAli/Persian-Text-Sentiment-Bert-V1')
+tokenizer = AutoTokenizer.from_pretrained('SeyedAli/Persian-Text-Sentiment-Bert-V1',add_special_token=True)
 def Sentiment(text):
     pipline = pipeline(task="text-classification", model=model, tokenizer=tokenizer)