Update app.py

app.py

@@ -6,8 +6,17 @@ from huggingface_hub import login
 import os
 import shutil
 
-# Authenticate with Hugging Face
-login(os.getenv("HF_TOKEN"))
+from huggingface_hub import login
+
+# Add this before model initialization
+hf_token = os.getenv("HF_TOKEN")
+login(token=hf_token, add_to_git_credential=True)
+
+# Then create model with explicit token
+model = HfApiModel(
+    "mistralai/Mixtral-8x7B-Instruct-v0.1",
+    token=hf_token  # Pass token explicitly
+)
 
 def analyze_data(csv_file, additional_notes=""):
     # Clear previous figures
@@ -16,7 +25,7 @@ def analyze_data(csv_file, additional_notes=""):
     os.makedirs('./figures', exist_ok=True)
 
     # Initialize model and agent
-    model = HfApiModel("meta-llama/Llama-3.1-70B-Instruct")
+    model = HfApiModel("mistralai/Mixtral-8x7B-Instruct-v0.1")  # Best overall
     agent = CodeAgent(
     tools=[],
     model=model,