Updated app.py

app.py

@@ -29,18 +29,20 @@ import shutil
 from dotenv import load_dotenv
 from huggingface_hub import login
 
+db_name = "vector_db"
+folder = "my-knowledge-base/"
 
-def process_files(files):
-    db_name = "vector_db"
-    folder = "my-knowledge-base/"
+load_dotenv()
+HF_TOKEN = os.getenv("HF_TOKEN")
+if HF_TOKEN is None:
+    raise ValueError("HF_TOKEN is not set. Check your .env file.")
+login(token=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)
 
-    load_dotenv()
-    HF_TOKEN = os.getenv("HF_TOKEN")
-    login(token=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)
 
     processed_files = []
@@ -210,6 +212,9 @@ with gr.Blocks(css=css) as ui:
     question_btn.click(chat, inputs=[question], outputs= [answer])
     visualise_btn.click(visualise_data, inputs=[result], outputs=[plot_2d,plot_3d])
 
+
+
 # Launch Gradio app
 if __name__ == "__main__":
+    
     ui.launch(share=True)