Update app.py

app.py

@@ -99,15 +99,8 @@ class BasicAgent:
             max_new_tokens=512,
             temperature=0.0,
         )
-        
         self.agent = CodeAgent(model=self, tools=[], add_base_tools=True)
         
-        # self.agent = CodeAgent(
-        #     model=model, 
-        #     tools=[],
-        #     add_base_tools=True
-        # )
-        
     def _serialize_messages(self, messages):
         prompt = []
         for m in messages:
@@ -117,7 +110,7 @@ class BasicAgent:
             prompt.append(f"{role}: {text}")
         return "\n".join(prompt)
 
-    def __call__(self, question: str) -> str:
+    def generate(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
 
         allowed = {"max_new_tokens", "temperature", "top_k", "top_p"}
@@ -138,6 +131,8 @@ class BasicAgent:
             "content": [{"type": "text", "text": response}],
         }
 
+    __call__ = generate
+
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,