Update model-test.py

model-test.py

@@ -1,11 +1,12 @@
-from transformers import BertTokenizer, BertForSequenceClassification
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
 from transformers import pipeline
 
-finbert = BertForSequenceClassification.from_pretrained('yiyanghkust/finbert-tone',num_labels=3)
-tokenizer = BertTokenizer.from_pretrained('yiyanghkust/finbert-tone')
 
-nlp = pipeline("sentiment-analysis", model=finbert, tokenizer=tokenizer)
+tokenizer = AutoTokenizer.from_pretrained("Alanhau/test")
+model = AutoModelForSequenceClassification.from_pretrained("Alanhau/test")
 
-sentences = ["Shit, Limit down!"]
+nlp = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
+
+sentences = ["hate you!"]
 results = nlp(sentences)
-print(results)  #LABEL_0: neutral; LABEL_1: positive; LABEL_2: negative
+print(results)