diff --git a/requirements.txt b/requirements.txt
index af2b6e6..2cc6d29 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,5 +4,6 @@ numpy<2.0.0
 pylint
 pyinstaller
 google-api-python-client
+pytest
 torchvision
 torch==2.2.1
\ No newline at end of file
diff --git a/src/test_local.py b/src/test_local.py
index bdf7bb1..9187a1f 100644
--- a/src/test_local.py
+++ b/src/test_local.py
@@ -180,3 +180,5 @@ def test_infer_method(self):
 
 if __name__ == "__main__":
     unittest.main()
+
+