Views
No views yet
khaledb12s/Arabic_Sentiment_Analysis_v21from transformers import AutoTokenizer, AutoModelForSequenceClassification
2import torch
3
4model_name = "khaledb12s/Arabic_Sentiment_Analysis_v2"
5
6tokenizer = AutoTokenizer.from_pretrained(model_name)
7model = AutoModelForSequenceClassification.from_pretrained(model_name)
8
9text = "مرحبا"
10inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True)
11outputs = model(**inputs)
12
13pred = torch.argmax(outputs.logits, dim=1).item()
14label = "positive" if pred == 1 else "negative"
15print(label)
16