helinivan commited on
Commit
1f668e3
·
1 Parent(s): 3f0f584

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -46,7 +46,7 @@ tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
46
  model = AutoModelForSequenceClassification.from_pretrained(MODEL_PATH)
47
 
48
  text = "We deden een man een nacht in een vat met cola en nu is hij dood"
49
- tokenized_text = tokenizer([preprocess_data(text)], padding=True, truncation=True, max_length=512, return_tensors="pt")
50
  output = model(**tokenized_text)
51
  probs = output.logits.softmax(dim=-1).tolist()[0]
52
  confidence = max(probs)
@@ -58,7 +58,7 @@ results = {"is_sarcastic": prediction, "confidence": confidence}
58
  Output:
59
 
60
  ```
61
- {'is_sarcastic': 1, 'confidence': 0.9999982118606567}
62
  ```
63
 
64
  ## Performance
 
46
  model = AutoModelForSequenceClassification.from_pretrained(MODEL_PATH)
47
 
48
  text = "We deden een man een nacht in een vat met cola en nu is hij dood"
49
+ tokenized_text = tokenizer([preprocess_data(text)], padding=True, truncation=True, max_length=256, return_tensors="pt")
50
  output = model(**tokenized_text)
51
  probs = output.logits.softmax(dim=-1).tolist()[0]
52
  confidence = max(probs)
 
58
  Output:
59
 
60
  ```
61
+ {'is_sarcastic': 1, 'confidence': 0.8915400505065918}
62
  ```
63
 
64
  ## Performance