Update app.py

app.py

@@ -30,9 +30,9 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 from smolagents import Tool, CodeAgent, DuckDuckGoSearchTool, FinalAnswerTool, HfApiModel
 
 
-class CodeLlamaTool(Tool):
-    name = "code_llama_tool"
-    description = "Solves reasoning/code questions using Meta Code Llama 7B Instruct"
+class QwenChatTool(Tool):
+    name = "qwen_chat_tool"
+    description = "Solves reasoning/code questions using Qwen1.5-7B-Chat"
 
     inputs = {
         "question": {
@@ -43,7 +43,7 @@ class CodeLlamaTool(Tool):
     output_type = "string"
 
     def __init__(self):
-        self.model_id = "codellama/CodeLlama-7b-Instruct-hf"
+        self.model_id = "Qwen/Qwen1.5-7B-Chat"
         token = os.getenv("HF_TOKEN")
 
         self.tokenizer = AutoTokenizer.from_pretrained(self.model_id, token=token)
@@ -113,7 +113,6 @@ Answer:"""
             return f"Error executing code:\n{traceback.format_exc()}"
 
 
-
 #from smolagents import Tool
 #from langchain_community.document_loaders import WikipediaLoader
 
@@ -287,7 +286,7 @@ class BasicAgent:
         video_transcription_tool = VideoTranscriptionTool()
 
         # ✅ New Llama Tool
-        code_llama_tool = CodeLlamaTool()
+        qwen_chat_tool = QwenChatTool()
 
         system_prompt = f"""
 You are my general AI assistant. Your task is to answer the question I asked.
@@ -306,7 +305,7 @@ If the answer is a comma-separated list, apply the above rules for each element
                 keywords_extract_tool, speech_to_text_tool,
                 visit_webpage_tool, final_answer_tool,
                 parse_excel_to_json, video_transcription_tool,
-                code_llama_tool  # 🔧 Add here
+                qwen_chat_tool  # 🔧 Add here
             ],
             add_base_tools=True
         )