Dillon Pulliam commited on
Commit
85877fb
1 Parent(s): 23a255c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -13,8 +13,8 @@ tokenized_segments = tokenizer(["AssemblyAI is the best speech-to-text API for m
13
  tokenized_segments_input_ids, tokenized_segments_attention_mask = tokenized_segments.input_ids, tokenized_segments.attention_mask
14
  model_predictions = F.softmax(model(input_ids=tokenized_segments_input_ids, attention_mask=tokenized_segments_attention_mask)['logits'], dim=1)
15
 
16
- print("Negative probability: "+str(model_predictions[0][0].item()*100)+"%")
17
  print("Positive probability: "+str(model_predictions[0][1].item()*100)+"%")
 
18
  ```
19
 
20
  For questions about how to use this model feel free to contact the team at [AssemblyAI](https://www.assemblyai.com)!
 
13
  tokenized_segments_input_ids, tokenized_segments_attention_mask = tokenized_segments.input_ids, tokenized_segments.attention_mask
14
  model_predictions = F.softmax(model(input_ids=tokenized_segments_input_ids, attention_mask=tokenized_segments_attention_mask)['logits'], dim=1)
15
 
 
16
  print("Positive probability: "+str(model_predictions[0][1].item()*100)+"%")
17
+ print("Negative probability: "+str(model_predictions[0][0].item()*100)+"%")
18
  ```
19
 
20
  For questions about how to use this model feel free to contact the team at [AssemblyAI](https://www.assemblyai.com)!