Muddassar commited on
Commit
ee6e30e
1 Parent(s): d83353e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -7,7 +7,7 @@ This model is a fine-tuned version of the Longformer base model for sentiment an
7
  - Model Type: Longformer
8
  - Task: Sentiment Analysis
9
  - Training Data: 4000 customer support tickets (Approx 1000 for each class)
10
- - Number of Parameters: {num_params}
11
 
12
  ## Performance
13
  - Overall Accuracy: 74.84%
@@ -32,6 +32,7 @@ weighted avg 0.76 0.75 0.73 477
32
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
33
  import torch
34
 
 
35
  model = AutoModelForSequenceClassification.from_pretrained("{hf_model_name}")
36
  tokenizer = AutoTokenizer.from_pretrained("{hf_model_name}")
37
 
@@ -48,4 +49,4 @@ print(f"Predicted sentiment: {sentiment_map[prediction]}")
48
 
49
 
50
  ## License
51
- [Specify the license for your model]
 
7
  - Model Type: Longformer
8
  - Task: Sentiment Analysis
9
  - Training Data: 4000 customer support tickets (Approx 1000 for each class)
10
+ - Number of Parameters: 149M
11
 
12
  ## Performance
13
  - Overall Accuracy: 74.84%
 
32
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
33
  import torch
34
 
35
+ hf_model_name= 'Muddassar/longformer-base-sentiment-5-classes'
36
  model = AutoModelForSequenceClassification.from_pretrained("{hf_model_name}")
37
  tokenizer = AutoTokenizer.from_pretrained("{hf_model_name}")
38
 
 
49
 
50
 
51
  ## License
52
+ MIT