diff --git a/CTAI_model/data_set/__init__.py b/CTAI_model/data_set/__init__.py
new file mode 100644
index 0000000..8b13789
--- /dev/null
+++ b/CTAI_model/data_set/__init__.py
@@ -0,0 +1 @@
+