Update app.py

app.py

@@ -109,9 +109,9 @@ def process_image(api_key, image_input, unit):
 def main(api_key, text_input="", image_input=None, unit):
     SYS_PROMPT = read('system_prompt.txt')
     if text_input and image_input is None:
-        return process_text(text_input,unit)
+        return process_text(api_key, text_input, unit)
     elif image_input is not None:
-        return process_image(image_input,unit)
+        return process_image(api_key, image_input, unit)
 
 with gr.Blocks(theme='shivi/calm_seafoam', css=css, title="Medster - Medical Diagnostic Assistant") as iface:
     with gr.Accordion(""):