Update README.md

README.md

@@ -1,3 +1,21 @@
----
-license: mit
----
+---
+license: mit
+---
+
+```python
+from transformers import AutoTokenizer, AutoModelForSequenceClassification, TextClassificationPipeline
+import re
+
+pipeline = TextClassificationPipeline(
+    model=AutoModelForSequenceClassification.from_pretrained("virtual-human-chc/prot_bert_bfd_membrane"),
+    tokenizer=AutoTokenizer.from_pretrained("virtual-human-chc/prot_bert_bfd_membrane"),
+    device=0
+)
+
+sequences_Example = ["M A K S K N H T A H N Q T R K A H R N G I K K P K T Y K Y P S L K G V D P K F R R N H K H A L H G T A K A L A A A K K",
+                     "M G L P V S W A P P A L W V L G C C A L L L S L W A L C T A C R R P E D A V A P R K R A R R Q R A R L Q G S A T A A E A S L L R R T H L C S L S K S D T R L H E L H R G P R S S R A L R P A S M D L L R P H W L E V S R D I T G P Q A A P S A F P H Q E L P R A L P A A A A T A G C A G L E A T Y S N V G L A A L P G V S L A A S P V V A E Y A R V Q K R K G T H R S P Q E P Q Q G K T E V T P A A Q V D V L Y S R V C K P K R R D P G P T T D P L D P K G Q G A I L A L A G D L A Y Q T L P L R A L D V D S G P L E N V Y E S I R E L G D P A G R S S T C G A G T P P A S S C P S L G R G W R P L P A S L P"]
+
+sequences_Example = [re.sub(r"[UZOB]", "X", sequence) for sequence in sequences_Example]
+
+print(pipeline(sequences_Example))
+```