Update app.py

app.py

@@ -28,32 +28,31 @@ class BasicAgent:
         print(f"Agent returning fixed answer: {fixed_answer}")
 
 
-        # model = HfApiModel(
-        #     max_tokens=2096,
-        #     temperature=0.0,
-        #     model_id="google/gemma-2b-it",# it is possible that this model may be overloaded
-        #     custom_role_conversions=None,
-        # )
-
-        # with open("prompts.yaml", 'r') as stream:
-        #     prompt_templates = yaml.safe_load(stream)
-
-        # agent = CodeAgent(
-        #     model=model,
-        #     # tools=[DuckDuckGoSearchTool()],
-        #     tools=[],
-        #     max_steps=5,
-        #     verbosity_level=1,
-        #     grammar=None,
-        #     planning_interval=None,
-        #     name=None,
-        #     description=None,
-        #     prompt_templates=prompt_templates
-        # )
-
-        # agent_answer = agent.run(question)
-
-        # print(f"Agent Answer: {agent_answer}")
+        model = TransformersModel(
+            model_id="microsoft/phi-2",   # ✅ small, reliable
+            max_new_tokens=1024,
+            temperature=0.0,
+        )
+
+        with open("prompts.yaml", 'r') as stream:
+            prompt_templates = yaml.safe_load(stream)
+
+        agent = CodeAgent(
+            model=model,
+            # tools=[DuckDuckGoSearchTool()],
+            tools=[],
+            max_steps=5,
+            verbosity_level=1,
+            grammar=None,
+            planning_interval=None,
+            name=None,
+            description=None,
+            prompt_templates=prompt_templates
+        )
+
+        agent_answer = agent.run(question)
+
+        print(f"Agent Answer: {agent_answer}")
 
         return fixed_answer