Update app.py

app.py

@@ -68,7 +68,8 @@ def process_text(text_input, unit):
     elif text_input and endpoints == "GOOGLE":
         genai.configure(api_key=api_key)
         model = genai.GenerativeModel(model_name=model_name),
-        response = model.generate_content(f" You are a experienced {unit} doctor." + SYS_PROMPT + f"Could you solve {text_input}?")
+        prompt = f" You are a experienced {unit} doctor." + SYS_PROMPT + f"Could you solve {text_input}?",
+        response = model.generate_content(prompt)
         return response.text
     return ""
 
@@ -104,7 +105,8 @@ def process_image(image_input, unit):
     elif image_input is not None and endpoints == "GOOGLE":
         genai.configure(api_key=api_key)
         model = genai.GenerativeModel(model_name=model_name),
-        response = model.generate_content([f" You are a experienced {unit} doctor." + SYS_PROMPT + "Help me understand what is in this picture and analysis.", image_input])
+        prompt = f" You are a experienced {unit} doctor." + SYS_PROMPT + "Help me understand what is in this picture and analysis.",
+        response = model.generate_content([prompt, image_input],request_options={"timeout": 60})
         return response.text
 
 def main(api_key="", text_input="", image_input=None, unit=""):