Update app.py

app.py

@@ -23,6 +23,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 #Load environment variables
 load_dotenv()
 
+import os
 import io
 import contextlib
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
@@ -59,10 +60,7 @@ class ZephyrToolCallingAgentTool(Tool):
             temperature=0.2
         )
 
-    def _run(self, question: str) -> str:
-        return self.run(question)
-
-    def run(self, question: str) -> str:
+    def forward(self, question: str) -> str:
         prompt = f"""You are a helpful assistant. Use code to solve questions that involve calculations.
 If code is needed, return a block like <tool>code</tool>. End your answer with <final>answer</final>.