Govi2020 commited on
Commit
e383c51
·
verified ·
1 Parent(s): c2e9676

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import AutoTokenizer, BertForSequenceClassification
4
  tokenizer = AutoTokenizer.from_pretrained("textattack/bert-base-uncased-yelp-polarity")
5
  model = BertForSequenceClassification.from_pretrained("textattack/bert-base-uncased-yelp-polarity")
6
 
7
- inputs = tokenizer("Hello, my dog is cute", return_tensors="pt")
8
 
9
  with torch.no_grad():
10
  logits = model(**inputs).logits
 
4
  tokenizer = AutoTokenizer.from_pretrained("textattack/bert-base-uncased-yelp-polarity")
5
  model = BertForSequenceClassification.from_pretrained("textattack/bert-base-uncased-yelp-polarity")
6
 
7
+ inputs = tokenizer("GEMINI AI Just got updated", return_tensors="pt")
8
 
9
  with torch.no_grad():
10
  logits = model(**inputs).logits