Update app.py

app.py

@@ -7,13 +7,14 @@ import torch
 model_name = "meta-llama/Meta-Llama-3.1-8B-Instruct"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto")
+# 환경 변수에서 토큰을 가져오기
+hf_token = os.environ.get("HF_TOKEN",None)
 
 #hugging face 로그인 (토큰 가져오기 위해서)
 from huggingface_hub import login
 import os
 
-# 환경 변수에서 토큰을 가져오기
-hf_token = os.environ.get("llama3.31-8b-token-new")
+
 
 # 토큰을 사용하여 로그인
 if hf_token: