revert back to hfapimodel

app.py

@@ -4,10 +4,10 @@ import os
 from smolagents import (
     tool,
     CodeAgent,
-TransformersModel,
+    TransformersModel,
     GradioUI,
     MultiStepAgent,
-    stream_to_gradio,
+    stream_to_gradio, HfApiModel,
 )
 from sqlalchemy import (
     create_engine,
@@ -120,10 +120,15 @@ def init_db(engine):
 if __name__ == "__main__":
     engine = create_engine("sqlite:///:localhost:")
     engine = init_db(engine)
-    model = TransformersModel(
-        # model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
-        device_map="cuda",
-        model_id="meta-llama/Llama-3.2-3B-Instruct"
+    #Not working at the moment
+    # model = TransformersModel(
+    #     # model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
+    #     device_map="cuda",
+    #     model_id="meta-llama/Llama-3.2-3B-Instruct"
+    # )
+    model = HfApiModel(
+          model_id="meta-llama/Llama-3.2-3B-Instruct",
+        token=os.getenv("my_first_agents_hf_tokens")
     )
 
     agent = CodeAgent(