gyesibiney commited on
Commit
22085d2
·
1 Parent(s): 5f5fd58

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -12,7 +12,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
12
  # Create a sentiment analysis pipeline
13
  sentiment = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
14
 
15
- # Create a dictionary to map sentiment labels to binary values
16
  sentiment_label_mapping = {
17
  "LABEL_1": "positive",
18
  "LABEL_0": "negative",
 
12
  # Create a sentiment analysis pipeline
13
  sentiment = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
14
 
15
+ # Create a dictionary to map sentiment labels to positive and negative strings
16
  sentiment_label_mapping = {
17
  "LABEL_1": "positive",
18
  "LABEL_0": "negative",