Synnove commited on
Commit
7d4421c
·
verified ·
1 Parent(s): d12d4c3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -36,7 +36,14 @@ class BasicAgent:
36
 
37
  class BasicCodeAgent:
38
  def __init__(self):
39
- model = InferenceClientModel()
 
 
 
 
 
 
 
40
  self.agent = CodeAgent(tools=[DuckDuckGoSearchTool()], model=model)
41
  print("BasicCodeAgent initialized.")
42
  def __call__(self, question: str) -> str:
 
36
 
37
  class BasicCodeAgent:
38
  def __init__(self):
39
+ #model = InferenceClientModel() #Cause of
40
+
41
+ model = HfApiModel(
42
+ max_tokens=2096,
43
+ temperature=0.5,
44
+ model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
45
+ custom_role_conversions=None,
46
+ )
47
  self.agent = CodeAgent(tools=[DuckDuckGoSearchTool()], model=model)
48
  print("BasicCodeAgent initialized.")
49
  def __call__(self, question: str) -> str: