--- a
+++ b/lungs/__init__.py
@@ -0,0 +1 @@
+from lungs.main import train, predict
\ No newline at end of file