Update backend.py
Browse files- 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.
|
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:
|