Update app.py

app.py

@@ -46,6 +46,8 @@ model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud/',#
 custom_role_conversions=None,
 )
 
+model = LiteLLMModel(model_id="gemini/gemini-2.0-flash-lite",
+                    api_key=os.getenv(key="AIzaSyCVUZhQm0DLyZnlYjpqldo-LSLOt3H9C6I"))
 
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)