Updated app.py

app.py

@@ -28,18 +28,19 @@ from random import randint
 import shutil
 from dotenv import load_dotenv
 
-db_name = "vector_db"
-folder = "my-knowledge-base/"
-
-load_dotenv()
-HF_TOKEN = os.getenv("HF_TOKEN")
-MODEL_NAME = "mistralai/Mistral-7B-Instruct-v0.3"
-tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME , token=HF_TOKEN)
-model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, device_map="auto",use_auth_token=HF_TOKEN)
 
 def process_files(files):
-    os.makedirs(folder, exist_ok=True)
+    db_name = "vector_db"
+    folder = "my-knowledge-base/"
 
+    load_dotenv()
+    HF_TOKEN = os.getenv("HF_TOKEN")
+    MODEL_NAME = "mistralai/Mistral-7B-Instruct-v0.3"
+    tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME , token=HF_TOKEN)
+    model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, device_map="auto",use_auth_token=HF_TOKEN)
+
+    os.makedirs(folder, exist_ok=True)
+    
     processed_files = []
     for file in files:
         file_path = os.path.join(folder, os.path.basename(file))  # Get filename