Ujeshhh commited on
Commit
8336431
·
verified ·
1 Parent(s): 3cabb2f

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 model for text generation
4
- llm_pipeline = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct", device_map="auto")
5
 
6
  def analyze_spending_pattern(df):
7
  prompt = f"""
 
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"""