thisisdev commited on
Commit
b296805
·
verified ·
1 Parent(s): b9d147d

Update backend.py

Browse files
Files changed (1) hide show
  1. backend.py +1 -1
backend.py CHANGED
@@ -14,7 +14,7 @@ class InvoicePipeline:
14
 
15
  def __init__(self, paths):
16
  self._paths = paths
17
- self._llm = GoogleGenerativeAI(model="gemini-1.5-flash", google_api_key=api_key)
18
  self._prompt_template = self._get_default_prompt_template()
19
  # This funcition will help in extracting and run the code, and will produce a dataframe for us
20
  def run(self) -> pd.DataFrame:
 
14
 
15
  def __init__(self, paths):
16
  self._paths = paths
17
+ self._llm = GoogleGenerativeAI(model="gemini-1.0-pro", google_api_key=api_key)
18
  self._prompt_template = self._get_default_prompt_template()
19
  # This funcition will help in extracting and run the code, and will produce a dataframe for us
20
  def run(self) -> pd.DataFrame: