Update app.py

app.py

@@ -1,7 +1,7 @@
 import os
 import torch
 import faiss
-from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
+from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig, Accelerate
 from sentence_transformers import SentenceTransformer
 from datasets import load_dataset
 import gradio as gr
@@ -11,10 +11,10 @@ hf_api_key = os.getenv('HF_API_KEY')
 
 # 모델 ID 및 토크나이저 설정
 model_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
-tokenizer = AutoTokenizer.from_pretrained(model_id, use_auth_token=hf_api_key)
+tokenizer = AutoTokenizer.from_pretrained(model_id, token=hf_api_key)
 model = AutoModelForCausalLM.from_pretrained(
     model_id,
-    use_auth_token=hf_api_key,
+    token=hf_api_key,
     torch_dtype=torch.bfloat16,
     device_map="auto",
     quantization_config=BitsAndBytesConfig(