Updated app.py

app.py

@@ -26,12 +26,15 @@ import plotly.express as px
 import matplotlib.pyplot as plt
 from random import randint
 import shutil
+from dotenv import load_dotenv
 
 db_name = "vector_db"
 folder = "my-knowledge-base/"
 
-MODEL_NAME = "mistralai/Mistral-7B-Instruct"  # Example: Mistral-7B
-tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
+load_dotenv()
+HF_TOKEN = os.getenv("HUGGINGFACE_TOKEN")
+MODEL_NAME = "mistralai/Mistral-7B-Instruct"
+tokenizer = AutoTokenizer.from_pretrained(model_name, token=HF_TOKEN)
 model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, device_map="auto")
 
 def process_files(files):