Update README.md

README.md

@@ -35,13 +35,13 @@ The model contains three versions:
   from transformers import AutoTokenizer, AutoModelForCausalLM
   
   tokenizer_test = AutoTokenizer.from_pretrained(
-      model_path,
-      token = 'xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx'  # Your access key of hugging face
+      model_path = "xxxxxxxxx", # The local path where the model is located
+      token = "xxxxxxxxx" # Your access key of hugging face
   ) 
   model_test = AutoModelForCausalLM.from_pretrained(
-      model_path,
-      device_map='auto',
-      token = 'xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx'  # Your access key of hugging face
+      model_path = "xxxxxxxxx", # The local path where the model is located
+      device_map = "auto",
+      token = "xxxxxxxxx" # Your access key of hugging face
   )
   
   def complete_chat(system, prompt, model, tokenizer):
@@ -72,7 +72,7 @@ The model contains three versions:
       return summary
 
   if __name__ == "__main__":
-    genes = "Your Gene Set Separated by comma (,)!"
+    genes = "xxxxxxxxx" # Your private gene set that is separated by comma (,)!
     result = llama(genes)
     print(result)
 ```