Update app.py

app.py

@@ -9,6 +9,7 @@ import io
 import contextlib
 import sys
 import traceback
+import os
 
 @tool
 def parse_height_from_text(
@@ -123,7 +124,7 @@ model = OpenAIServerModel(
 # Replace all calls to HfApiModel
 llm_model = LiteLLMModel(
     model_id="gemini/gemini-2.0-flash", # you can see other model names here: https://cloud.google.com/vertex-ai/generative-ai/docs/learn/models. It is important to prefix the name with "gemini/"
-    api_key=userdata.get("GEM_TOKEN"),
+    api_key=os.environ.get('GEM_TOKEN'),
     max_tokens=8192
 )