Update app.py

app.py

@@ -4,6 +4,8 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+from google.colab import userdata
+
 
 from Gradio_UI import GradioUI
 
@@ -50,7 +52,6 @@ model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',
 custom_role_conversions=None,
 )
 '''
-from google.colab import userdata
 model = LiteLLMModel(model_id="gemini/gemini-2.0-flash-lite", api_key=userdata.get('AIzaSyBzW2rm1bR317GGQAWFHnCxVxxpIomKOHw'))