From febdd79793fe8f0b8725fc94cec63fdc68706591 Mon Sep 17 00:00:00 2001
From: Dhruv Gautam <108053562+dhruvji@users.noreply.github.com>
Date: Mon, 7 Jul 2025 13:46:42 -0700
Subject: [PATCH] Update README.md

---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/README.md b/README.md
index ebd7b18d..7e0681c6 100644
--- a/README.md
+++ b/README.md
@@ -95,7 +95,7 @@ in the TOML file:
 
 
 ```bash
-state tx infer --output $HOME/state/test/ --output_dir /path/to/model/ --checkpoint /path/to/model/final.ckpt --adata /path/to/anndata/processed.h5 --pert_col gene --embed_key X_hvg
+state tx infer --output $HOME/state/test/ --model_dir /path/to/model/ --checkpoint /path/to/model/final.ckpt --adata /path/to/anndata/processed.h5 --pert_col gene --embed_key X_hvg
 ```
 
 Here, `/path/to/model/` is the folder downloaded from [HuggingFace](https://huggingface.co/arcinstitute).