Update README.md

README.md

@@ -1,5 +1,4 @@
 ---
-# 1. Metadata Block
 license: mit
 library_name: pytorch-lightning
 pipeline_tag: tabular-classification
@@ -7,7 +6,7 @@ tags:
 - biology
 - genomics
 datasets:
-- Genentech/human-chromhmm-fullstack-data
+- Genentech/human-atac-catlas-data
 base_model:
 - Genentech/enformer-model
 ---
@@ -15,11 +14,11 @@ base_model:
 # human-chromhmm-fullstack-model
 
 ## Model Description
-This model is a multi-class classifier trained to classify genomic DNA sequences into one of 16 classes. It was trained by fine-tuning the Enformer model using the `grelu` library on top of the human ChromHMM fullstack annotation dataset.
+This model is a multi-task classifier trained to predict the binary accessibility of genomic DNA sequences in 204 cell types. It was trained by fine-tuning the Enformer model using the `grelu` library on top of the human ChromHMM fullstack annotation dataset.
 
 - **Architecture:** Fine-tuned Enformer
 - **Input:** Genomic sequences (hg38)
-- **Output:** Probability across 16 class labels.
+- **Output:** Probability of accessibility in 204 cell types.
 
 ## Repository Content
 1. `model.ckpt`: The trained model weights and hyperparameters (PyTorch Lightning checkpoint).
@@ -34,7 +33,7 @@ from grelu.lightning import LightningModel
 from huggingface_hub import hf_hub_download
 
 ckpt_path = hf_hub_download(
-    repo_id="Genentech/human-chromhmm-fullstack-model", 
+    repo_id="Genentech/human-atac-catlas-model", 
     filename="model.ckpt"
 )