Update app.py

app.py

@@ -8,12 +8,17 @@ import pandas as pd
 import time
 import re
 from markdownify import markdownify
-from smolagents import Tool, DuckDuckGoSearchTool, CodeAgent, WikipediaSearchTool, LiteLLMModel
+from smolagents import Tool, DuckDuckGoSearchTool, CodeAgent, WikipediaSearchTool, LiteLLMModel, HfApiModel
 
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
-
+model_qwen = HfApiModel(
+                model_id='Qwen/Qwen-VL-Chat',
+                max_tokens=2048,   # Reasonable default for this model
+                temperature=0.2,   # You can adjust this based on how creative you want answers to be
+                custom_role_conversions=None
+            )
 
 class DownloadTaskAttachmentTool(Tool):
     name = "download_file"
@@ -95,7 +100,7 @@ class VisitWebpageTool(Tool):
 class BasicAgent:
     def __init__(self):
         self.agent = CodeAgent(
-            model=LiteLLMModel(model_id="openrouter/meta-llama/llama-4-maverick:free", api_key=os.getenv("OPENROUTER_KEY")),
+            model= model_qwen,
             tools=[DuckDuckGoSearchTool(), WikipediaSearchTool(), VisitWebpageTool(), DownloadTaskAttachmentTool()],
             add_base_tools=True,
             additional_authorized_imports=['pandas','numpy','csv','subprocess', 'exec']