gyesibiney commited on
Commit
e8cce98
·
1 Parent(s): 22f96fd

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -14,8 +14,8 @@ 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": 1, # Positive
18
- "LABEL_0": 0, # Negative
19
  }
20
 
21
  # Define a request body model
 
14
 
15
  # Create a dictionary to map sentiment labels to binary values
16
  sentiment_label_mapping = {
17
+ "LABEL_1": "positive",
18
+ "LABEL_0": "negative",
19
  }
20
 
21
  # Define a request body model