Update app.py

app.py

@@ -48,8 +48,29 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         
         
 
-        model = InferenceClientModel(model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud')
-        agent = CodeAgent(tools=[DuckDuckGoSearchTool()], model=model)
+        model = HfApiModel(
+            max_tokens=2096,
+            temperature=0.5,
+            model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',
+            #model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
+            custom_role_conversions=None,
+        )
+        
+        with open("prompts.yaml", 'r') as stream:
+            prompt_templates = yaml.safe_load(stream)
+            
+        # We're creating our CodeAgent
+        agent = CodeAgent(
+            model=model,
+            tools=[final_answer, DuckDuckGoSearchTool()], 
+            max_steps=6,
+            verbosity_level=1,
+            grammar=None,
+            planning_interval=None,
+            name=None,
+            description=None,
+            prompt_templates=prompt_templates
+        )
         
         #agent = BasicAgent()
     except Exception as e: