Create README.md

README.md

@@ -0,0 +1,20 @@
+# BERT-Large-Uncased for Sentiment Analysis
+This model is a fine-tuned version of [bert-large-uncased](https://huggingface.co/bert-large-uncased) originally released in ["BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding"](https://arxiv.org/abs/1810.04805) and trained on the [Stanford Sentiment Treebank v2 (SST2)](https://nlp.stanford.edu/sentiment/); part of the [General Language Understanding Evaluation (GLUE)](https://gluebenchmark.com) benchmark. This model was been fine-tuned by the team at [AssemblyAI](https://www.assemblyai.com) and is released with the [corresponding blog post]().  
+
+## Usage
+To download and utilize this model for sentiment analysis please execute the following:
+```python
+import torch.nn.functional as F 
+from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassification
+tokenizer = AutoTokenizer.from_pretrained("assemblyai/bert-large-uncased-sst2") 
+model = AutoModelForSequenceClassification.from_pretrained("assemblyai/bert-large-uncased-sst2")
+
+tokenized_segments = tokenizer(["AssemblyAI is the best speech-to-text API for modern developers with performance being second to none!"], return_tensors="pt", padding=True, truncation=True)
+tokenized_segments_input_ids, tokenized_segments_attention_mask = tokenized_segments.input_ids, tokenized_segments.attention_mask
+model_predictions = F.softmax(model(input_ids=tokenized_segments_input_ids, attention_mask=tokenized_segments_attention_mask)['logits'], dim=1)
+
+print("Negative probability: "+str(model_predictions[0][0].item()*100)+"%")
+print("Positive probability: "+str(model_predictions[0][1].item()*100)+"%")
+```
+
+For questions about how to use this model feel free to contact the team at [AssemblyAI](https://www.assemblyai.com)!