Vadim Borisov commited on
Commit
2eeab69
1 Parent(s): 6c841bb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -27,7 +27,7 @@ pipeline_tag: text-classification
27
  # BERT-based Sentiment Classification Mode
28
 
29
  ## Model Details
30
- - **Model Name:** tabularisai/robust_sentiment_analysis
31
  - **Base Model:** bert-base-uncased
32
  - **Task:** Text Classification (Sentiment Analysis)
33
  - **Language:** English
@@ -62,7 +62,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
62
  import torch
63
 
64
  # Load model and tokenizer
65
- model_name = "tabularisai/robust_sentiment_analysis"
66
  tokenizer = AutoTokenizer.from_pretrained(model_name)
67
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
68
 
@@ -133,7 +133,7 @@ The model demonstrates strong performance across various sentiment categories. H
133
  env.allowLocalModels = false;
134
  env.useCDN = true;
135
 
136
- const MODEL_NAME = 'tabularisai/robust_sentiment_analysis';
137
 
138
  function softmax(arr) {
139
  const max = Math.max(...arr);
 
27
  # BERT-based Sentiment Classification Mode
28
 
29
  ## Model Details
30
+ - **Model Name:** tabularisai/robust-sentiment-analysis
31
  - **Base Model:** bert-base-uncased
32
  - **Task:** Text Classification (Sentiment Analysis)
33
  - **Language:** English
 
62
  import torch
63
 
64
  # Load model and tokenizer
65
+ model_name = "tabularisai/robust-sentiment-analysis"
66
  tokenizer = AutoTokenizer.from_pretrained(model_name)
67
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
68
 
 
133
  env.allowLocalModels = false;
134
  env.useCDN = true;
135
 
136
+ const MODEL_NAME = 'tabularisai/robust-sentiment-analysis';
137
 
138
  function softmax(arr) {
139
  const max = Math.max(...arr);