Views
No views yet
1import torch
2from transformers import AutoTokenizer, AutoModelForSequenceClassification
3
4tokenizer = AutoTokenizer.from_pretrained("harixn/IN-finbert")
5model = AutoModelForSequenceClassification.from_pretrained("harixn/IN-finbert")
6
7text = "The stock price of XYZ surged today."
8inputs = tokenizer(text, return_tensors="pt")
9outputs = model(**inputs)
10
11# Get probabilities
12probs = torch.softmax(outputs.logits, dim=1)
13print("Probabilities:", probs)
14
15# Get predicted class
16pred_class = torch.argmax(probs, dim=1).item()
17classes = ["negative", "neutral", "positive"]
18print("Predicted class:", classes[pred_class])pytorch_model.bin: Trained model weightsconfig.json: Model configurationvocab.txt, tokenizer_config.json, special_tokens_map.json, tokenizer.json: Tokenizer filesFinBERT: Sentiment Analysis Model for Indian Stock Market, harixn, 2025