Update app.py

app.py

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