Update second.py

second.py

@@ -6,13 +6,14 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 from langchain_groq import ChatGroq
 from langchain.prompts import ChatPromptTemplate
 
-# 環境変数からAPIキーを設定
+# 環境変数からHugging Faceのトークンを取得
+os.environ['HF_TOKEN'] = os.getenv('HUGGING_FACE_TOKEN')
 os.environ["GROQ_API_KEY"] = os.getenv("GROQ_API_KEY")
 
 # Hugging Faceのモデルとトークナイザーをロード
 model_id = "meta-llama/Meta-Llama-3-8B-Instruct"
-tokenizer = AutoTokenizer.from_pretrained(model_id)
-model = AutoModelForCausalLM.from_pretrained(model_id)
+tokenizer = AutoTokenizer.from_pretrained(model_id, use_auth_token=os.getenv('HF_TOKEN'))
+model = AutoModelForCausalLM.from_pretrained(model_id, use_auth_token=os.getenv('HF_TOKEN'))
 
 # Groqを使ったLangChainのChatGroqインスタンスを初期化
 chat = ChatGroq(model_name=model_id, tokenizer=tokenizer)