Update README.md
Browse files
README.md
CHANGED
@@ -70,3 +70,45 @@ The following hyperparameters were used during training:
|
|
70 |
- Transformers 4.47.1
|
71 |
- Pytorch 2.5.1+cu121
|
72 |
- Tokenizers 0.21.0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
70 |
- Transformers 4.47.1
|
71 |
- Pytorch 2.5.1+cu121
|
72 |
- Tokenizers 0.21.0
|
73 |
+
|
74 |
+
### How to use:
|
75 |
+
```python
|
76 |
+
model_path = "ehab215/egyptian_sentiment_analysis"
|
77 |
+
tokenizer = AutoTokenizer.from_pretrained(model_path)
|
78 |
+
model = AutoModelForSequenceClassification.from_pretrained(model_path)
|
79 |
+
|
80 |
+
# Ensure model is on GPU if available
|
81 |
+
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
82 |
+
model.to(device)
|
83 |
+
|
84 |
+
# Step 2: Prepare test examples
|
85 |
+
examples = [
|
86 |
+
"والله الفرح امبارح كان تحفة، الناس كلها كانت مبسوطة والأكل كان جامد وكله تمام",
|
87 |
+
"مبروك يا معلم على الشغل الجديد! انت تستاهل كل خير والله، ربنا يوفقك",
|
88 |
+
"النهاردة الجو عادي، لا حر ولا برد، يعني معتدل",
|
89 |
+
"المحل اللي تحت البيت بيفتح من الصبح لحد بالليل، بيبيع كل حاجة",
|
90 |
+
"الزحمة في الشارع بقت مش طبيعية، مش عارف اروح شغلي في معاد، تعبت بجد",
|
91 |
+
"العربية خربانة من اسبوع والميكانيكي مش عارف يصلحها، ومصاريف كتير على الفاضي",
|
92 |
+
"الانترنت في البيت بطيء جداً وبيفصل كل شوية، اتصلت بالشركة مليون مرة ومفيش فايدة",
|
93 |
+
]
|
94 |
+
|
95 |
+
# Tokenize the examples
|
96 |
+
inputs = tokenizer(examples, truncation=True, padding=True, return_tensors="pt", max_length=256)
|
97 |
+
inputs = {key: val.to(device) for key, val in inputs.items()}
|
98 |
+
|
99 |
+
# Step 3: Make predictions
|
100 |
+
with torch.no_grad():
|
101 |
+
outputs = model(**inputs)
|
102 |
+
logits = outputs.logits
|
103 |
+
predictions = torch.argmax(logits, dim=-1).cpu().numpy()
|
104 |
+
|
105 |
+
# Step 4: Interpret results
|
106 |
+
label_map = {0: "negative", 1: "neutral", 2: "positive"}
|
107 |
+
predicted_labels = [label_map[p] for p in predictions]
|
108 |
+
|
109 |
+
# Display results
|
110 |
+
for text, label in zip(examples, predicted_labels):
|
111 |
+
print(f"Text: {text}")
|
112 |
+
print(f"Predicted Sentiment: {label}")
|
113 |
+
print("-" * 50)
|
114 |
+
```
|