Update app.py
Browse files
app.py
CHANGED
@@ -3,8 +3,8 @@ from transformers import TFBertForSequenceClassification, BertTokenizer
|
|
3 |
import tensorflow as tf
|
4 |
|
5 |
# Load model and tokenizer from your HF model repo
|
6 |
-
model = TFBertForSequenceClassification.from_pretrained("
|
7 |
-
tokenizer = BertTokenizer.from_pretrained("
|
8 |
|
9 |
def classify_sentiment(text):
|
10 |
inputs = tokenizer(text, return_tensors="tf", padding=True, truncation=True)
|
|
|
3 |
import tensorflow as tf
|
4 |
|
5 |
# Load model and tokenizer from your HF model repo
|
6 |
+
model = TFBertForSequenceClassification.from_pretrained("shrish191/sentiment-bert")
|
7 |
+
tokenizer = BertTokenizer.from_pretrained("shrish191/sentiment-bert")
|
8 |
|
9 |
def classify_sentiment(text):
|
10 |
inputs = tokenizer(text, return_tensors="tf", padding=True, truncation=True)
|