gradio requirement.txt'

app.py

@@ -1,12 +1,8 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 import os
-from smolagents import tool, CodeAgent, HfApiModel, GradioUI  # type: ignore
+from smolagents import tool, CodeAgent, HfApiModel, GradioUI
 
-# testing teste
-"""
-For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
-"""
 client = InferenceClient(
     "HuggingFaceH4/zephyr-7b-beta", token=os.getenv("my_first_agents_hf_tokens")
 )
@@ -34,14 +30,7 @@ def sql_engine(query: str) -> str:
     return output
 
 
-if __name__ == "__main__":
-    agent = CodeAgent(
-        tools=[sql_engine],
-        model=HfApiModel(
-            model_id="meta-llama/Meta-Llama-3.1-8B-Instruct",
-            token=os.getenv("my_first_agents_hf_tokens"),
-        ),
-    )
+def init_db():
 
     from sqlalchemy import (
         create_engine,
@@ -93,4 +82,14 @@ if __name__ == "__main__":
         },
     ]
     insert_rows_into_table(rows, receipts)
+
+
+if __name__ == "__main__":
+    agent = CodeAgent(
+        tools=[sql_engine],
+        model=HfApiModel(
+            model_id="meta-llama/Meta-Llama-3.1-8B-Instruct",
+            token=os.getenv("my_first_agents_hf_tokens"),
+        ),
+    )
     GradioUI(agent).launch()

requirements.txt

@@ -1,4 +1,5 @@
 huggingface_hub>=0.28.0
 smolagents==1.12.0
 python-dotenv==1.1.0
-sqlalchemy==2.0.40
+sqlalchemy==2.0.40
+gradio>0.0.0