ehab215 commited on
Commit
cacb183
·
verified ·
1 Parent(s): 81cd7ba

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +42 -0
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
+ ```