Update app.py

app.py

@@ -1,18 +1,17 @@
 #!/usr/bin/env python3
 
 import os
-import re
 import ast
 import operator
 import datetime
 import pytz
 import json
+import re
 import requests
-import yaml
 import pandas as pd
 import gradio as gr
 
-from smolagents import CodeAgent, HfApiModel, tool
+from smolagents import CodeAgent, TransformersModel, tool
 
 # -------------------------
 # Minimal tools
@@ -80,19 +79,16 @@ def get_current_time_in_timezone(timezone: str) -> str:
 # -------------------------
 prompt_templates = None
 try:
+    import yaml
     with open("prompts.yaml", "r") as fh:
         prompt_templates = yaml.safe_load(fh)
 except Exception:
     prompt_templates = None
 
 # -------------------------
-# HfApiModel + CodeAgent minimal
+# TransformersModel + CodeAgent minimal
 # -------------------------
-model = HfApiModel(
-    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
-    max_tokens=1024,
-    temperature=0.0
-)
+model = TransformersModel(model_id="HuggingFaceTB/SmolLM-135M-Instruct")
 
 code_agent = CodeAgent(
     model=model,