Views
No views yet
1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2import torch
3
4tokenizer = AutoTokenizer.from_pretrained("stocky-ai/fin-impact-xlmr")
5model = AutoModelForSequenceClassification.from_pretrained("stocky-ai/fin-impact-xlmr")
6
7def predict(text):
8 inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True)
9 with torch.no_grad():
10 return model(**inputs).logits.item() * 10
11
12print(predict("Central bank raised interest rates amid inflation concerns"))