Dillon Pulliam
commited on
Commit
•
c6a84d6
1
Parent(s):
e951196
Update README.md
Browse files
README.md
CHANGED
@@ -13,8 +13,8 @@ tokenized_segments = tokenizer(["How many hours does it take to fly from Califor
|
|
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("
|
17 |
-
print("
|
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("Duplicate probability: "+str(model_predictions[0][1].item()*100)+"%")
|
17 |
+
print("Non-duplicate 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)!
|