Update app.py

app.py

@@ -106,7 +106,7 @@ def process_image(api_key, image_input, unit):
         response = model.generate_content([prompt, image_input],request_options={"timeout": 60})
         return response.text
 
-def main(api_key="", text_input="", image_input=None, 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)
@@ -133,7 +133,7 @@ with gr.Blocks(theme='shivi/calm_seafoam', css=css, title="Medster - Medical Dia
         clear_btn = gr.ClearButton(output_box, value="🗑️ Clear") # Create a clear button
 
     # Set up the event listeners
-    submit_btn.click(main, inputs=[api_key, text_input, image_input, unit], outputs=output_box)
+    submit_btn.click(main, inputs=[unit, api_key, text_input, image_input], outputs=output_box)
 
     gr.Markdown(LICENSE)