Update app.py

app.py

@@ -6,10 +6,9 @@ from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
-# A tool that sums two numbers and returns the sum as an integer.
 @tool
 def my_custom_tool(arg1: int, arg2: int) -> int:
-    """A tool that sums two numbers.
+    """ A tool that sums two numbers and returns the sum as an integer.
     
     Args:
         arg1: the first number.
@@ -36,8 +35,7 @@ final_answer = FinalAnswerTool()
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
-    # model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
-    model_id='mistralai/Mistral-7B-Instruct-v0.2',
+    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
     custom_role_conversions=None,
 )