updating model peptriever_2023-06-23T16:07:24.508460

README.md

@@ -0,0 +1,27 @@
+---
+language: en
+license: mit
+datasets:
+- ronig/protein_binding_sequences
+---
+# Protein BiEncoder Bert Model
+
+Usage
+```python
+tokenizer = AutoTokenizer.from_pretrained("ronig/protein_biencoder")
+model = BiEncoder.from_pretrained("ronig/protein_biencoder")
+model.eval()
+
+peptide_sequence = "AAA"
+protein_sequence = "MMM"
+encoded_peptide = tokenizer.encode_plus(peptide_sequence, return_tensors='pt')
+encoded_protein = tokenizer.encode_plus(protein_sequence, return_tensors='pt')
+
+with torch.no_grad():
+    peptide_output = model.forward1(encoded_peptide)
+    protein_output = model.forward2(encoded_protein)
+
+print("distance: ", torch.norm(peptide_output - protein_output, p=2))
+```
+
+Model checkpint: `peptriever_2023-06-23T16:07:24.508460`