gnanechaithu commited on
Commit
3dcf645
·
verified ·
1 Parent(s): dcee602

changed the model

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -1,5 +1,5 @@
1
 
2
- from smolagents import (CodeAgent, HfApiModel, tool, GradioUI)
3
  import os
4
 
5
  HF_TOKEN = os.environ.get("HFRTOKEN")
@@ -25,7 +25,11 @@ def tell_location()->any:
25
  return timezone
26
 
27
 
28
- model = HfApiModel(model_id="deepseek-ai/DeepSeek-R1-Distill-Qwen-32B",token=HF_TOKEN, custom_role_conversions=None, provider="hf-inference")
 
 
 
 
29
 
30
  agent = CodeAgent(model=model, tools=[greet_someone, tell_location])
31
 
 
1
 
2
+ from smolagents import (CodeAgent, InferenceClientModel, tool, GradioUI)
3
  import os
4
 
5
  HF_TOKEN = os.environ.get("HFRTOKEN")
 
25
  return timezone
26
 
27
 
28
+ # model = HfApiModel(model_id="deepseek-ai/DeepSeek-R1-Distill-Qwen-32B",token=HF_TOKEN, custom_role_conversions=None, provider="hf-inference")
29
+
30
+ model = InferenceClientModel(model_id="deepseek-ai/DeepSeek-Prover-V2-671B",
31
+ provider="together",
32
+ token=HF_TOKEN)
33
 
34
  agent = CodeAgent(model=model, tools=[greet_someone, tell_location])
35