MichaelHuang commited on
Commit
b78f8b6
1 Parent(s): 2219dcb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +29 -0
README.md CHANGED
@@ -14,5 +14,34 @@ Urdu sentiment analysis dataset is from [mirfan899](https://github.com/mirfan899
14
  ## Usage
15
  ### example:
16
  ```python
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
17
 
18
  ```
 
14
  ## Usage
15
  ### example:
16
  ```python
17
+ import torch
18
+ from transformers import AutoTokenizer, AutoModelForSequenceClassification
19
+
20
+ # Load the model and tokenizer
21
+ tokenizer = AutoTokenizer.from_pretrained("google/muril-base-cased")
22
+ model = AutoModelForSequenceClassification.from_pretrained("MichaelHuang/muril_base_cased_urdu_sentiment")
23
+
24
+ # with open('/Users/michaelhuang/Dropbox/University of Chicago/2023 Spring/DIGS 30032/Final project/Urdu News Authorship Classifier/urdu/INP/IR/04_07_27_2023/آرامکو میں ریاستی ملکیت کا 4فیصد سنابیل انویسٹمنٹ کو منتقل کرنے کا اعلانتازترین.txt', 'r', encoding='utf-8') as file:
25
+ # text = file.read()
26
+
27
+ # Define the input text
28
+ text = '''
29
+ لیکن مسٹر پوتن نے یہ بھی کہا کہ یہ منصوبہ اسی وقت پیش کیا جا سکتا ہے جب لوگ 'مغرب اور کیئو میں' اس کے لیے تیار ہوں۔
30
+ روسی رہنما نے منگل کو ماسکو میں چینی صدر شی جن پنگ سے ملاقات کی جس میں روس یوکرین جنگ اور دونوں ممالک کے درمیان تعلقات پر تبادلہ خیال کیا گیا۔
31
+ گذشتہ ماہ شائع ہونے والے چین کے منصوبے میں واضح طور پر روس سے یوکرین چھوڑنے کا مطالبہ نہیں کیا گیا ہے۔
32
+ '''
33
+
34
+ # Tokenize the input text
35
+ inputs = tokenizer(text, return_tensors='pt')
36
+
37
+ # Make a prediction
38
+ outputs = model(**inputs)
39
+ predicted_class = torch.argmax(outputs.logits).item()
40
+
41
+ # Print the predicted class
42
+ if predicted_class == 1:
43
+ print('Positive')
44
+ else:
45
+ print('Negative')
46
 
47
  ```