Views
No views yet
1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2import torch
3
4model_name = "ShivamSharma8168/hindi-sentiment-model"
5
6tokenizer = AutoTokenizer.from_pretrained(model_name)
7model = AutoModelForSequenceClassification.from_pretrained(model_name)
8
9labels = ["negative", "neutral", "positive"]
10
11def predict_sentiment(text):
12 inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True)
13 outputs = model(**inputs)
14 pred = torch.argmax(outputs.logits, dim=1).item()
15 return labels[pred]
16
17print(predict_sentiment("यह फ़िल्म बहुत अच्छी है")) # positive
18print(predict_sentiment("मुझे यह पसंद नहीं आया")) # negative