Ujeshhh commited on
Commit
8bc6866
·
verified ·
1 Parent(s): f93fe1a

Update analysis.py

Browse files
Files changed (1) hide show
  1. analysis.py +2 -2
analysis.py CHANGED
@@ -1,7 +1,7 @@
1
  from transformers import pipeline
2
 
3
- # Load a free Google LLM model for text generation
4
- llm_pipeline = pipeline("text-generation", model="google/gemma-7b", device_map="auto")
5
 
6
  def analyze_spending_pattern(df):
7
  prompt = f"""
 
1
  from transformers import pipeline
2
 
3
+ # Use Mistral-7B instead of Google Gemma (No login required)
4
+ llm_pipeline = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct", device_map="auto")
5
 
6
  def analyze_spending_pattern(df):
7
  prompt = f"""