model change

app.py

@@ -2,6 +2,20 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 import os
 from smolagents import tool, CodeAgent, HfApiModel, GradioUI
+from sqlalchemy import (
+    create_engine,
+    MetaData,
+    Table,
+    Column,
+    String,
+    Integer,
+    Float,
+    insert,
+    inspect,
+    text,
+)
+
+engine = create_engine("sqlite:///:memory:")
 
 
 @tool
@@ -28,20 +42,6 @@ def sql_engine(query: str) -> str:
 
 def init_db():
 
-    from sqlalchemy import (
-        create_engine,
-        MetaData,
-        Table,
-        Column,
-        String,
-        Integer,
-        Float,
-        insert,
-        inspect,
-        text,
-    )
-
-    engine = create_engine("sqlite:///:memory:")
     metadata_obj = MetaData()
 
     def insert_rows_into_table(rows, table, engine=engine):
@@ -81,10 +81,11 @@ def init_db():
 
 
 if __name__ == "__main__":
+    init_db()
     agent = CodeAgent(
         tools=[sql_engine],
         model=HfApiModel(
-            model_id="meta-llama/Meta-Llama-3.1-8B-Instruct",
+            model_id="HuggingFaceTB/SmolLM2-1.7B-Instruct",
             token=os.getenv("my_first_agents_hf_tokens"),
         ),
     )