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)
|