Update app.py

app.py

@@ -23,7 +23,7 @@ class BasicAgent:
     def __init__(self):
         self.model = TransformersModel(
             model_id="Qwen/Qwen2.5-Coder-7B-Instruct",
-            max_new_tokens=1024,
+            max_new_tokens=768,
             temperature=0.1,
         )
 
@@ -76,10 +76,11 @@ class BasicAgent:
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         fixed_answer = "This is a default answer."
-        print(f"Agent returning fixed answer: {fixed_answer}")
+        # print(f"Agent returning fixed answer: {fixed_answer}")
 
 
-        if question == "Given this table defining * on the set S = {a, b, c, d, e}\n\n|*|a|b|c|d|e|\n|---|---|---|---|---|---|\n|a|a|b|c|b|d|\n|b|b|c|a|e|c|\n|c|c|a|b|b|a|\n|d|b|e|b|e|d|\n|e|d|b|a|d|c|\n\nprovide the subset of S involved in any possible counter-examples that prove * is not commutative. Provide your answer as a comma separated list of the elements in the set in alphabetical order.":
+        # if question == "Given this table defining * on the set S = {a, b, c, d, e}\n\n|*|a|b|c|d|e|\n|---|---|---|---|---|---|\n|a|a|b|c|b|d|\n|b|b|c|a|e|c|\n|c|c|a|b|b|a|\n|d|b|e|b|e|d|\n|e|d|b|a|d|c|\n\nprovide the subset of S involved in any possible counter-examples that prove * is not commutative. Provide your answer as a comma separated list of the elements in the set in alphabetical order.":
+        if " image " not in question and " video " not in question:
             agent_answer = self.agent.run(question)
         else:
             agent_answer = fixed_answer