ayoub-dhibi commited on
Commit
2adc06f
·
verified ·
1 Parent(s): 2483cd9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -39,11 +39,14 @@ final_answer = FinalAnswerTool()
39
  # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
40
  # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
41
 
 
 
 
42
 
43
  model = LiteLLMModel(
44
  model_id="gemini/gemini-2.0-flash-lite-001",
45
  temperature=0.7,
46
- api_key=userdata.get("GEMINI_KEY")
47
  )
48
  '''
49
  model = HfApiModel(
 
39
  # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
40
  # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
41
 
42
+ import os
43
+
44
+ api_key = os.getenv("GEMINI_KEY")
45
 
46
  model = LiteLLMModel(
47
  model_id="gemini/gemini-2.0-flash-lite-001",
48
  temperature=0.7,
49
+ api_key=api_key
50
  )
51
  '''
52
  model = HfApiModel(