fatsam commited on
Commit
858c3d7
·
verified ·
1 Parent(s): a2cd7c6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -36,7 +36,8 @@ final_answer = FinalAnswerTool()
36
  model = HfApiModel(
37
  max_tokens=2096,
38
  temperature=0.5,
39
- model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
 
40
  custom_role_conversions=None,
41
  )
42
 
@@ -58,4 +59,8 @@ agent = CodeAgent(
58
  prompt_templates=prompt_templates
59
  )
60
 
 
 
 
 
61
  GradioUI(agent).launch()
 
36
  model = HfApiModel(
37
  max_tokens=2096,
38
  temperature=0.5,
39
+ # model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
40
+ model_id='mistralai/Mistral-7B-Instruct-v0.2',
41
  custom_role_conversions=None,
42
  )
43
 
 
59
  prompt_templates=prompt_templates
60
  )
61
 
62
+ print("Agent running:", prompt)
63
+ result = agent.run(prompt)
64
+ print("Agent result:", result)
65
+
66
  GradioUI(agent).launch()