Update app.py

app.py

@@ -3,18 +3,19 @@ import pandas as pd
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 
-# 모델과 토크나이저 로드
+# model, tokenizer 셋팅
 model_name = "meta-llama/Meta-Llama-3.1-8B"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto")
 
+#hugging face 로그인 (토큰 가져오기 위해서)
 from huggingface_hub import login
 import os
 
-# 환경 변수에서 토큰을 가져옵니다
-hf_token = os.environ.get("HF_TOKEN")
+# 환경 변수에서 토큰을 가져오기
+hf_token = os.environ.get("llama3.31-8b-token-new")
 
-# 토큰을 사용하여 로그인합니다
+# 토큰을 사용하여 로그인
 if hf_token:
     login(token=hf_token)
 else: