Update app.py
Browse files
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 |
|