sunnysharma20 commited on
Commit
2e4e294
·
verified ·
1 Parent(s): 05f88c9

Update backend.py

Browse files
Files changed (1) hide show
  1. backend.py +1 -1
backend.py CHANGED
@@ -10,7 +10,7 @@ os.environ["GOOGLE_API_KEY"]="AIzaSyC8W6-q0Db2200MDE3LpsVZPrkzXZUUuW0"
10
 
11
  class InvoicePipeline:
12
 
13
- def __init__(self, paths:List[str]):
14
  self._paths = paths
15
  self ._llm = GoogleGenerativeAI(model="gemini-1.5-pro", google_api_key=api_key)
16
  self._prompt_template = self._get_default_prompt_template()
 
10
 
11
  class InvoicePipeline:
12
 
13
+ def __init__(self, paths):
14
  self._paths = paths
15
  self ._llm = GoogleGenerativeAI(model="gemini-1.5-pro", google_api_key=api_key)
16
  self._prompt_template = self._get_default_prompt_template()