Update README.md

README.md

@@ -44,7 +44,12 @@ Use the code below to get started with the model.
 
 First, download the necessary libraries.
 ```bash
-pip install femr torch
+# Create Python 3.10 environment
+conda create --name ehrshot_env python=3.10 -y
+conda activate ehrshot_env
+
+# Install requirements
+pip install torch==2.1.1 femr==0.2.0 datasets==2.15.0 flash_attn==2.3.6 transformers==4.35.2
 ```
 
 Second, run the following Python script to run inference on a single patient: