Update analysis.py

analysis.py

@@ -1,9 +1,21 @@
 from transformers import pipeline
+from huggingface_hub import login
+import pandas as pd
 
-# Use Mistral-7B instead of Google Gemma (No login required)
-llm_pipeline = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct", device_map="auto")
+# 🔹 OPTIONAL: Authenticate if using a gated/private model
+# login(token="your_huggingface_token")  # Uncomment and replace if needed
+
+# ✅ Use a Free Model (Mistral-7B-v0.1 OR Gemma-2B)
+MODEL_NAME = "mistralai/Mistral-7B-v0.1"  # Publicly available
+# MODEL_NAME = "google/gemma-2b"  # Alternative (smaller but open-access)
+
+# Load the text generation model
+llm_pipeline = pipeline("text-generation", model=MODEL_NAME, device_map="auto")
 
 def analyze_spending_pattern(df):
+    """
+    Analyze the user's spending behavior.
+    """
     prompt = f"""
     Here is the user's spending data:
     {df.to_string(index=False)}
@@ -14,6 +26,9 @@ def analyze_spending_pattern(df):
     return response[0]['generated_text']
 
 def get_financial_advice(df):
+    """
+    Provide personalized financial recommendations.
+    """
     prompt = f"""
     Given the following transaction history:
     {df.to_string(index=False)}