diff --git a/.gitignore b/.gitignore
index e82c9d2..c6206b0 100644
--- a/.gitignore
+++ b/.gitignore
@@ -87,9 +87,6 @@ venv.bak/
 .dmypy.json
 dmypy.json
 
-# Batdetect Models [Include]
-!bat_detect/models/*.pth.tar
-
 # Model artifacts
 *.png
 *.jpg
@@ -98,6 +95,9 @@ dmypy.json
 *.json
 plots/*
 
+# Batdetect Models [Include]
+!bat_detect/models/*.pth.tar
+
 # Model experiments
 experiments/*
 
diff --git a/bat_detect/models/Net2DFast_UK_same.pth.tar b/bat_detect/models/Net2DFast_UK_same.pth.tar
new file mode 100644
index 0000000..e3704b7
Binary files /dev/null and b/bat_detect/models/Net2DFast_UK_same.pth.tar differ
diff --git a/pyproject.toml b/pyproject.toml
index f9a1adc..696460e 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -12,7 +12,7 @@ dependencies = [
     "pandas",
     "scikit-learn",
     "scipy",
-    "torch",
+    "torch<2",
     "torchaudio",
     "torchvision",
     "click",