vilarin commited on
Commit
4089b9f
·
verified ·
1 Parent(s): f14ae84

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -106,7 +106,7 @@ def process_image(api_key, image_input, unit):
106
  response = model.generate_content([prompt, image_input],request_options={"timeout": 60})
107
  return response.text
108
 
109
- def main(api_key, text_input="", image_input=None, unit):
110
  SYS_PROMPT = read('system_prompt.txt')
111
  if text_input and image_input is None:
112
  return process_text(api_key, text_input, unit)
 
106
  response = model.generate_content([prompt, image_input],request_options={"timeout": 60})
107
  return response.text
108
 
109
+ def main(unit="", api_key="", text_input="", image_input=None):
110
  SYS_PROMPT = read('system_prompt.txt')
111
  if text_input and image_input is None:
112
  return process_text(api_key, text_input, unit)