Update app.py

app.py

@@ -15,8 +15,10 @@ import os
 api_key = os.getenv('NNSIGHT_API_KEY')
 CONFIG.set_default_api_key(api_key)
 
+access_token = os.environ['HF_TOKEN_WRITE']
+
 # Load the Language Model
-llama = LanguageModel("meta-llama/Meta-Llama-3.1-8B", token=os.environ['HF_TOKEN_WRITE'])
+llama = LanguageModel("meta-llama/Meta-Llama-3.1-8B", token=access_token)
 
 #placeholder for reset
 prompts_with_probs = pd.DataFrame(