diff --git a/inference.py b/inference.py
index 5c38a6c..98912de 100644
--- a/inference.py
+++ b/inference.py
@@ -29,6 +29,8 @@ def legacy_compatibility(args, checkpoint):
     elif args.firenet_legacy:
         args.legacy_norm = True
         final_activation = ''
+    else:
+        return args, checkpoint
     # Make compatible with Henri saved models
     if not isinstance(checkpoint.get('config', None), ConfigParser) or args.e2vid or args.firenet_legacy:
         checkpoint = make_henri_compatible(checkpoint, final_activation)
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..a4cf91f
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,7 @@
+torch
+torchvision
+matplotlib
+opencv-python
+tqdm
+pandas
+h5py