Update README.md

README.md

@@ -17,6 +17,7 @@ Use `https://phenogemini.org` to generate a prompt that includes similar patient
 
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
+import torch
 
 model_name = "DISCOStudy/PhenoGemini"
 
@@ -29,8 +30,7 @@ base = AutoModelForCausalLM.from_pretrained(
     torch_dtype="auto",
     device_map="auto",
     trust_remote_code=True
-)
-
+).eval()
 ```
 
 ## Logit Masking (Gene-Token Subspace Ranking)