Upload README.md with huggingface_hub

README.md

@@ -45,22 +45,22 @@ Refer to the [paper](https://www.biorxiv.org/content/10.64898/2026.06.03.729735)
 
 ### Usage
 
-Please install `esm` from GitHub (a PyPI release is coming soon):
+Please install `esm` from PyPI:
 
 ```
-pip install esm@git+https://github.com/Biohub/esm.git@main
+pip install esm
 ```
 
 You can fold your first protein with:
 
 ```py
-from transformers.models.esmfold2.modeling_esmfold2 import ESMFold2Model
+from transformers.models.esmfold2.modeling_esmfold2 import EsmFold2Model
 
 # Ubiquitin (PDB 1UBQ)
 sequence = "MQIFVKTLTGKTITLEVEPSDTIENVKAKIQDKEGIPPDQQRLIFAGKQLEDGRTLSDYNIQKESTLHLVLRLRGG"
 
 # optionally use "biohub/ESMFold2"
-model = ESMFold2Model.from_pretrained("biohub/ESMFold2-Fast").cuda().eval()
+model = EsmFold2Model.from_pretrained("biohub/ESMFold2-Fast", device_map="auto").eval()
 output = model.infer_protein(sequence, num_loops=3, num_sampling_steps=50)
 
 print(f"pLDDT mean: {float(output['plddt'].mean()):.3f}, pTM: {float(output['ptm'].mean()):.3f}")
@@ -72,12 +72,12 @@ You can also fold complex biomolecules — proteins, DNA/RNA (with modified resi
 from esm.models.esmfold2 import (
     DNAInput,
     ESMFold2InputBuilder,
+    EsmFold2Model,
     LigandInput,
     Modification,
     ProteinInput,
     StructurePredictionInput,
 )
-from transformers.models.esmfold2.modeling_esmfold2 import ESMFold2Model
 
 HHAI_SEQ = (
     "MIEIKDKQLTGLRFIDLFAGLGGFRLALESCGAECVYSNEWDKYAQEVYEMNFGEKPEGDITQVNEKTIPDH"
@@ -87,7 +87,7 @@ HHAI_SEQ = (
     "YKVHPSTSQAYKQFGNSVVINVLQYIAYNIGSSLNFKPY"
 )
 
-model = ESMFold2Model.from_pretrained("biohub/ESMFold2").cuda().eval()
+model = EsmFold2Model.from_pretrained("biohub/ESMFold2", device="cuda").eval()
 
 spi = StructurePredictionInput(
     sequences=[