Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -21,8 +21,8 @@ GO-GPT is a decoder-only transformer model for predicting Gene Ontology (GO) ter
|
|
| 21 |
|
| 22 |
1. Clone the repository:
|
| 23 |
```bash
|
| 24 |
-
git clone https://github.com/
|
| 25 |
-
cd gogpt
|
| 26 |
```
|
| 27 |
|
| 28 |
2. Run the inference notebook or use Python directly:
|
|
@@ -33,7 +33,7 @@ sys.path.insert(0, "src")
|
|
| 33 |
from gogpt import GOGPTPredictor
|
| 34 |
|
| 35 |
# Load from HuggingFace (downloads ~4GB on first run)
|
| 36 |
-
predictor = GOGPTPredictor.from_pretrained("
|
| 37 |
|
| 38 |
# Predict GO terms
|
| 39 |
predictions = predictor.predict(
|
|
@@ -42,9 +42,9 @@ predictions = predictor.predict(
|
|
| 42 |
)
|
| 43 |
|
| 44 |
print(predictions)
|
| 45 |
-
# {
|
| 46 |
-
#
|
| 47 |
-
#
|
| 48 |
```
|
| 49 |
|
| 50 |
## Model Architecture
|
|
|
|
| 21 |
|
| 22 |
1. Clone the repository:
|
| 23 |
```bash
|
| 24 |
+
git clone https://github.com/bowang-lab/BioReason-Pro
|
| 25 |
+
cd BioReason-Pro/gogpt
|
| 26 |
```
|
| 27 |
|
| 28 |
2. Run the inference notebook or use Python directly:
|
|
|
|
| 33 |
from gogpt import GOGPTPredictor
|
| 34 |
|
| 35 |
# Load from HuggingFace (downloads ~4GB on first run)
|
| 36 |
+
predictor = GOGPTPredictor.from_pretrained("wanglab/gogpt")
|
| 37 |
|
| 38 |
# Predict GO terms
|
| 39 |
predictions = predictor.predict(
|
|
|
|
| 42 |
)
|
| 43 |
|
| 44 |
print(predictions)
|
| 45 |
+
# {"MF": ["GO:0003674", "GO:0005488", ...],
|
| 46 |
+
# "BP": ["GO:0008150", "GO:0008152", ...],
|
| 47 |
+
# "CC": ["GO:0005575", "GO:0110165", ...]}
|
| 48 |
```
|
| 49 |
|
| 50 |
## Model Architecture
|