Merge pull request #482 from Atharva-Phatak/update-torch

modify requirements

Author: Atharva-Phatak

.pre-commit-config.yaml

@@ -20,19 +20,4 @@ repos:
     hooks:
       - id: black
         exclude: ^tests/
-        args: [ --safe, --quiet ]
-  - repo: https://gitlab.com/pycqa/flake8
-    rev: 3.8.1
-    hooks:
-      - id: flake8
-        additional_dependencies: [ flake8-blind-except, flake8-bugbear, flake8-comprehensions,
-                                     flake8-debugger, flake8-docstrings, flake8-isort, flake8-polyfill, flake8-pytest,
-                                     flake8-quotes, yesqa ]
-        exclude: ^tests/|docs/source/conf.py/|^examples/
-
-  - repo: https://github.com/PyCQA/bandit
-    rev: '1.7.0' # Update me!
-    hooks:
-      - id: bandit
-        args: [--skip, "B101"]
-        exclude: ^tests/|^examples/|^docs/
+        args: [ --safe, --quiet ]

requirements.txt

@@ -1,7 +1,7 @@
-numpy==1.20.3
-pandas==1.2.5
-torch==1.9.0
-torchvision==0.10.0
-albumentations==1.0.0
-torchmetrics==0.5.0
-gradio==2.2.13
+numpy
+pandas
+torch==2.0.1
+torchvision
+albumentations
+torchmetrics
+gradio