Updated app.py

app.py

@@ -27,6 +27,7 @@ import matplotlib.pyplot as plt
 from random import randint
 import shutil
 from dotenv import load_dotenv
+from huggingface_hub import login
 
 
 def process_files(files):
@@ -35,12 +36,13 @@ def process_files(files):
 
     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)
 
     os.makedirs(folder, exist_ok=True)
-    
+
     processed_files = []
     for file in files:
         file_path = os.path.join(folder, os.path.basename(file))  # Get filename