Update app.py

app.py

@@ -12,12 +12,14 @@ models = {
         "mistralai/Mistral-7B-Instruct-v0.3",
         device_map="auto",
         torch_dtype=torch.bfloat16,  # Use bfloat16
+        use_auth_token=True,
         token=hf_token  # Add authentication here
     ),
     "BICORP/Lake-1-Advanced": AutoModelForCausalLM.from_pretrained(
         "BICORP/Lake-1-Advanced",
         device_map="auto",
         torch_dtype=torch.bfloat16,  # Use bfloat16
+        use_auth_token=True,
         token=hf_token  # Add authentication here
     )
 }
@@ -25,10 +27,12 @@ models = {
 tokenizers = {
     "mistralai/Mistral-7B-Instruct-v0.3": AutoTokenizer.from_pretrained(
         "mistralai/Mistral-7B-Instruct-v0.3",
+        use_auth_token=True,
         token=hf_token  # Add authentication here
     ),
     "BICORP/Lake-1-Advanced": AutoTokenizer.from_pretrained(
         "BICORP/Lake-1-Advanced",
+        use_auth_token=True,
         token=hf_token  # Add authentication here
     )
 }