Update app.py

app.py

@@ -106,7 +106,7 @@ def chat_with_infinite_agent(message, lang, history):
 #Gradio UI
 css_path = os.path.join(os.path.dirname(__file__), "style.css")
 
-def api_predict(message, lang="English"):
+def api_predict(message: str, lang: str = "English") -> dict:
     history, _ = chat_with_infinite_agent(message, lang, [])
     return {"response": history[-1]["content"]}