AmelieSchreiber commited on
Commit
62b7d50
1 Parent(s): c9a0891

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -3
README.md CHANGED
@@ -15,6 +15,10 @@ pipeline_tag: zero-shot-classification
15
  # ESM-2 Sequence Classifier
16
  This is a small sequence classifier trained on synthetic data generate by GPT-4
17
  which classifies protein sequences into three categories `enzymes`, `transport_proteins`, and `structural_proteins`.
 
 
 
 
18
  To use the model, try running:
19
 
20
  ``` python3
@@ -81,6 +85,3 @@ predicted_class_ids = torch.argmax(logits, dim=-1)
81
  for sequence, predicted_class in zip(new_sequences, predicted_class_ids):
82
  print(f"Sequence: {sequence}, Predicted class: {predicted_class.item()}")
83
  ```
84
-
85
- This is trained using [facebook/esm2_t6_8M_UR50D](https://huggingface.co/facebook/esm2_t6_8M_UR50D), one of the
86
- Meta AI [ESM-2 models](https://huggingface.co/docs/transformers/model_doc/esm).
 
15
  # ESM-2 Sequence Classifier
16
  This is a small sequence classifier trained on synthetic data generate by GPT-4
17
  which classifies protein sequences into three categories `enzymes`, `transport_proteins`, and `structural_proteins`.
18
+ This is trained using [facebook/esm2_t6_8M_UR50D](https://huggingface.co/facebook/esm2_t6_8M_UR50D), one of the
19
+ Meta AI [ESM-2 models](https://huggingface.co/docs/transformers/model_doc/esm).
20
+
21
+ ## Using the Model
22
  To use the model, try running:
23
 
24
  ``` python3
 
85
  for sequence, predicted_class in zip(new_sequences, predicted_class_ids):
86
  print(f"Sequence: {sequence}, Predicted class: {predicted_class.item()}")
87
  ```