switch to llama llm 3

app.py

@@ -19,7 +19,6 @@ import spaces
 
 from dotenv import load_dotenv
 
-engine = create_engine("sqlite:///:memory:")
 load_dotenv()
 
 # What is the average each customer paid? Create a sql statement and invoke your sql_engine tool
@@ -55,7 +54,7 @@ def sql_engine_tool(
     return output
 
 
-def init_db():
+def init_db(engine):
 
     metadata_obj = MetaData()
 
@@ -95,10 +94,12 @@ def init_db():
     insert_rows_into_table(rows, receipts)
     with engine.begin() as conn:
         print("SELECT test", conn.execute(text("SELECT * FROM receipts")).fetchall())
+    return engine
 
 
 if __name__ == "__main__":
-    init_db()
+    engine = create_engine("sqlite:///:memory:")
+    engine = init_db(engine)
     model = HfApiModel(
         model_id="",
         token=os.getenv("my_first_agents_hf_tokens"),