Create README.md

README.md

@@ -0,0 +1,22 @@
+---
+license: apache-2.0
+base_model:
+- google-bert/bert-base-uncased
+---
+
+You can load the laff model using the following steps: 
+
+```
+from transformers import BertTokenizer, BertForSequenceClassification
+from base_models import  BinaryClassificationBertModel
+
+device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+
+base_model_name = "bert-base-uncased" 
+tokenizer = BertTokenizer.from_pretrained(base_model_name, torch_dtype=torch.float16)
+base_model = BertForSequenceClassification.from_pretrained(base_model_name, num_labels=1,torch_dtype=torch.float16)
+
+model = BinaryClassificationBertModel(base_model)
+model.load_state_dict(torch.load("laff_model/bert-base-laff.pth"))
+model.to(device)
+```