diff --git a/main.py b/main.py
index 7410768..f32ba5e 100644
--- a/main.py
+++ b/main.py
@@ -1,8 +1,10 @@
 from ui import GenerateUI
 from dotenv import load_dotenv
+from src.models import init_db
 
 load_dotenv()
 
 if __name__ == "__main__":
+    init_db()
     ui_generator = GenerateUI()
     ui_generator.launch_ui()