Update app.py

app.py

@@ -6,7 +6,7 @@ import pandas as pd
 
 # Import our custom tools from their modules
 from huggingface_hub import login
-from smolagents import CodeAgent, InferenceClientModel
+from smolagents import CodeAgent, InferenceClientModel, OpenAIServerModel
 from tools import web_search, visit_webpage, final_answer, go_back, close_popups, search_item_ctrl_f
 from retriever import load_guest_dataset
 from functools import lru_cache
@@ -42,6 +42,8 @@ class BasicAgent:
         
         # Create the model
         self.model = InferenceClientModel(model_id="Qwen/Qwen2.5-72B-Instruct")
+        # self.model = InferenceClientModel(model_id="Qwen/Qwen2.5-7B-Instruct") #smaller/faster
+        # self.model = OpenAIServerModel(model_id="gpt-4o")
 
         # Define your system prompt
         self.system_prompt = """You are a general AI assistant. I will ask you a question. Finish your answer with only: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string."""