I have deployed my fast api application in databricks apps and I have given keep alive timeout 1200.

Issue:

From databricks swagger I am getting 504 "upstream request timeout" after 2 mins while my api takes 3 min to respond. 

But in backend my task got complete. So it's clearly a databricks gateway timeout.

So I could not find any option in databricks to increase the timeout. If databricks apps doesnot have as such option then need to think twice as we have used it in a client production based on databricks suggestion. 

Any quick solution?