Update demo.py

demo.py

@@ -7,11 +7,7 @@ import os
 os.environ["CUBLAS_WORKSPACE_CONFIG"] = ":4096:8"
 import contextlib
 
-from transformers import get_tokenizer
-
-api_key=get_tokenizer().api_key
-
-
+api_key = os.environ.get("api_key")
 
 import torch