diff --git a/.gitignore b/.gitignore
index 760e3e8..c3247e2 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,4 +1,4 @@
 .ipynb_checkpoints
 .idea
-.tar.gz
-.pth
+*.tar.gz
+*.pth
diff --git a/demo_code/model_LinearRegression.pth b/demo_code/model_LinearRegression.pth
deleted file mode 100644
index 8fce4cc..0000000
Binary files a/demo_code/model_LinearRegression.pth and /dev/null differ
diff --git a/demo_code/model_LogsticRegression.pth b/demo_code/model_LogsticRegression.pth
deleted file mode 100644
index 18749fb..0000000
Binary files a/demo_code/model_LogsticRegression.pth and /dev/null differ