Update app.py

app.py

@@ -14,6 +14,7 @@ import numpy as np
 from transformers import AutoTokenizer, AutoModelForCausalLM
 #from torch.nn.functional import softmax
 
+from huggingface_hub import login
 
 # just for the sake of this demo, we use cloze prob to initialize the hypothesis 
 
@@ -55,11 +56,14 @@ def get_sim(x):
 
 
 import os
-print(os.getenv('HF_token'))
+#print(os.getenv('HF_token'))
 hf_api_token = os.getenv("HF_token")  # For sensitive secrets
-app_mode = os.getenv("APP_MODE")  # For public variables
+#app_mode = os.getenv("APP_MODE")  # For public variables
 
 
+access_token = hf_api_token
+print(login(token = access_token))
+
 
 tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-3.2-1B")
 model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-3.2-1B")