Update app.py
Browse files
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
|
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)
|