Stevenfunau commited on
Commit
be81370
·
verified ·
1 Parent(s): 87a0fba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -17,11 +17,12 @@ import yaml
17
  DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
18
  #--SM
19
  #model_id="https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud/"
20
- model_id="Qwen/Qwen2.5-Coder-32B-Instruct"
21
  #model=LiteLLMModel(model_id="deepseek-ai/DeepSeek-Coder-V2-Lite-Base", api_key=os.getenv('HF_API_KEY'))
 
22
 
23
  # Initialize the model
24
- model = HfApiModel(model_id)
25
  #model = HfApiModel("Qwen/Qwen2.5-Coder-32B-Instruct")
26
  #model = HfApiModel("deepseek-ai/DeepSeek-R1", provider="together", max_tokens=8096)
27
 
 
17
  DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
18
  #--SM
19
  #model_id="https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud/"
20
+ #model_id="Qwen/Qwen2.5-Coder-32B-Instruct"
21
  #model=LiteLLMModel(model_id="deepseek-ai/DeepSeek-Coder-V2-Lite-Base", api_key=os.getenv('HF_API_KEY'))
22
+ model = LiteLLMModel(model_id="gemini/gemini-2.0-flash-lite", api_key=os.getenv('GEMINI_API_KEY'))
23
 
24
  # Initialize the model
25
+ #model = HfApiModel(model_id)
26
  #model = HfApiModel("Qwen/Qwen2.5-Coder-32B-Instruct")
27
  #model = HfApiModel("deepseek-ai/DeepSeek-R1", provider="together", max_tokens=8096)
28