diff --git a/enviroment.yml b/enviroment.yml
index 1fdc005..0eee2bd 100644
--- a/enviroment.yml
+++ b/enviroment.yml
@@ -1,185 +1,24 @@
 name: ium
 channels:
+  - pytorch
   - anaconda
   - conda-forge
   - defaults
 dependencies:
-  - absl-py=1.0.0=pyhd8ed1ab_0
-  - alembic=1.7.7=pyhd8ed1ab_0
-  - appdirs=1.4.4=pyh9f0ad1d_0
-  - asn1crypto=1.5.1=pyhd8ed1ab_0
-  - astor=0.8.1=pyh9f0ad1d_0
-  - bottleneck=1.3.4=py37h3a130e4_1
-  - brotli=1.0.9=h8ffe710_7
-  - brotli-bin=1.0.9=h8ffe710_7
-  - brotlipy=0.7.0=py37hcc03f2d_1004
-  - ca-certificates=2022.5.18.1=h5b45459_0
-  - cached-property=1.5.2=hd8ed1ab_1
-  - cached_property=1.5.2=pyha770c72_1
-  - certifi=2022.5.18.1=py37h03978a9_0
-  - cffi=1.15.0=py37hd8e9650_0
-  - charset-normalizer=2.0.12=pyhd8ed1ab_0
-  - click=8.1.3=py37h03978a9_0
-  - cloudpickle=2.1.0=pyhd8ed1ab_0
-  - colorama=0.4.4=pyh9f0ad1d_0
-  - configparser=5.2.0=pyhd8ed1ab_0
-  - cryptography=37.0.1=py37h21b164f_0
-  - cycler=0.11.0=pyhd8ed1ab_0
-  - databricks-cli=0.12.1=pyhd8ed1ab_0
-  - docker-py=5.0.3=py37h03978a9_2
-  - docker-pycreds=0.4.0=py_0
-  - docopt=0.6.2=py_1
-  - entrypoints=0.4=pyhd8ed1ab_0
-  - flask=2.1.2=pyhd8ed1ab_1
-  - fonttools=4.33.3=py37hcc03f2d_0
-  - freetype=2.10.4=h546665d_1
-  - gast=0.5.3=pyhd8ed1ab_0
-  - gettext=0.19.8.1=ha2e2712_1008
-  - gitdb=4.0.9=pyhd8ed1ab_0
-  - gitpython=3.1.27=pyhd8ed1ab_0
-  - google-pasta=0.2.0=pyh8c360ce_0
-  - greenlet=1.1.2=py37hf2a7229_2
-  - grpcio=1.46.3=py37h04d2302_0
-  - gst-plugins-base=1.20.2=he07aa86_1
-  - gstreamer=1.20.2=hdff456e_1
-  - h5py=3.6.0=nompi_py37h2502845_100
-  - hdf5=1.12.1=nompi_h2a0e4a3_104
-  - icu=70.1=h0e60522_0
-  - idna=3.3=pyhd8ed1ab_0
-  - importlib-metadata=4.11.3=py37h03978a9_1
-  - importlib_metadata=4.11.3=hd8ed1ab_1
-  - importlib_resources=5.7.1=pyhd8ed1ab_1
-  - intel-openmp=2022.1.0=h57928b3_3787
-  - itsdangerous=2.1.2=pyhd8ed1ab_0
-  - jbig=2.1=h8d14728_2003
-  - jinja2=3.1.2=pyhd8ed1ab_0
-  - joblib=1.1.0=pyhd3eb1b0_0
-  - jpeg=9e=h8ffe710_1
-  - jsonpickle=1.5.1=pyhd8ed1ab_0
-  - kaggle=1.5.12=pyhd8ed1ab_4
-  - keras-applications=1.0.8=py_1
-  - keras-preprocessing=1.1.2=pyhd8ed1ab_0
-  - kiwisolver=1.4.2=py37h8c56517_1
-  - krb5=1.19.3=h1176d77_0
-  - lcms2=2.12=h2a16943_0
-  - lerc=3.0=h0e60522_0
-  - libblas=3.9.0=14_win64_mkl
-  - libbrotlicommon=1.0.9=h8ffe710_7
-  - libbrotlidec=1.0.9=h8ffe710_7
-  - libbrotlienc=1.0.9=h8ffe710_7
-  - libcblas=3.9.0=14_win64_mkl
-  - libclang=14.0.3=default_h77d9078_0
-  - libclang13=14.0.3=default_h77d9078_0
-  - libcurl=7.83.1=h789b8ee_0
-  - libdeflate=1.10=h8ffe710_0
-  - libffi=3.4.2=h8ffe710_5
-  - libglib=2.70.2=h3be07f2_4
-  - libiconv=1.16=he774522_0
-  - liblapack=3.9.0=14_win64_mkl
-  - libogg=1.3.4=h8ffe710_1
-  - libpng=1.6.37=h1d00b33_2
-  - libprotobuf=3.20.1=h7755175_0
-  - libssh2=1.10.0=h680486a_2
-  - libtiff=4.3.0=hc4061b1_3
-  - libvorbis=1.3.7=h0e60522_0
-  - libwebp=1.2.2=h57928b3_0
-  - libwebp-base=1.2.2=h8ffe710_1
-  - libxcb=1.13=hcd874cb_1004
-  - libzlib=1.2.11=h8ffe710_1014
-  - lz4-c=1.9.3=h8ffe710_1
-  - m2w64-gcc-libgfortran=5.3.0=6
-  - m2w64-gcc-libs=5.3.0=7
-  - m2w64-gcc-libs-core=5.3.0=7
-  - m2w64-gmp=6.1.0=2
-  - m2w64-libwinpthread-git=5.0.0.4634.697f757=2
-  - mako=1.2.0=pyhd8ed1ab_1
-  - markdown=3.3.7=pyhd8ed1ab_0
-  - markupsafe=2.1.1=py37hcc03f2d_1
-  - matplotlib=3.5.2=py37h03978a9_0
-  - matplotlib-base=3.5.2=py37h54234da_0
-  - mkl=2022.0.0=h0e2418a_796
-  - mlflow=1.26.0=py37hd3201af_0
-  - msys2-conda-epoch=20160418=1
-  - munch=2.5.0=py_0
-  - munkres=1.1.4=pyh9f0ad1d_0
-  - numexpr=2.8.0=mkl_py37h4782e91_2
-  - numpy=1.21.6=py37h2830a78_0
-  - openjpeg=2.4.0=hb211442_1
-  - openssl=1.1.1o=h8ffe710_0
-  - packaging=21.3=pyhd8ed1ab_0
-  - pandas=1.3.5=py37h6214cd6_0
-  - patsy=0.5.2=pyhd8ed1ab_0
-  - pcre=8.45=h0e60522_0
-  - pillow=9.1.1=py37h8675073_0
-  - pip=22.1.1=pyhd8ed1ab_0
-  - prometheus_client=0.14.1=pyhd8ed1ab_0
-  - prometheus_flask_exporter=0.20.1=pyhd8ed1ab_0
-  - protobuf=3.20.1=py37h03978a9_0
-  - pthread-stubs=0.4=hcd874cb_1001
-  - py-cpuinfo=8.0.0=pyhd8ed1ab_0
-  - pycparser=2.21=pyhd8ed1ab_0
-  - pymongo=4.1.1=py37hf2a7229_0
-  - pyopenssl=22.0.0=pyhd8ed1ab_0
-  - pyparsing=3.0.9=pyhd8ed1ab_0
-  - pyqt=5.15.4=py37hf2a7229_0
-  - pyqt5-sip=12.9.0=py37hf2a7229_0
-  - pysocks=1.7.1=py37h03978a9_5
-  - python=3.7.12=h7840368_100_cpython
-  - python-dateutil=2.8.2=pyhd8ed1ab_0
-  - python-slugify=6.1.2=pyhd8ed1ab_0
-  - python_abi=3.7=2_cp37m
-  - pytz=2022.1=pyhd8ed1ab_0
-  - pywin32=303=py37hcc03f2d_0
-  - pyyaml=6.0=py37hcc03f2d_4
-  - qt-main=5.15.3=h467ea89_1
-  - querystring_parser=1.2.4=py_0
-  - requests=2.27.1=pyhd8ed1ab_0
-  - sacred=0.8.2=pyhd8ed1ab_0
-  - scikit-learn=1.0.2=py37hf11a4ad_1
-  - scipy=1.7.3=py37hb6553fb_0
-  - seaborn=0.11.2=hd8ed1ab_0
-  - seaborn-base=0.11.2=pyhd8ed1ab_0
-  - setuptools=62.3.2=py37h03978a9_0
-  - sip=6.5.1=py37hf2a7229_2
-  - six=1.16.0=pyh6c4a22f_0
-  - smmap=3.0.5=pyh44b312d_0
-  - sqlalchemy=1.4.36=py37hcc03f2d_0
-  - sqlite=3.38.5=h8ffe710_0
-  - sqlparse=0.4.2=pyhd8ed1ab_0
-  - statsmodels=0.13.2=py37hec80d1f_0
-  - tabulate=0.8.9=pyhd8ed1ab_0
-  - tbb=2021.5.0=h2d74725_1
-  - tenacity=8.0.1=pyhd8ed1ab_0
-  - tensorboard=1.14.0=py37_0
-  - tensorflow=1.14.0=h1f41ff6_0
-  - tensorflow-base=1.14.0=py37hc8dfbb8_0
-  - tensorflow-estimator=1.14.0=py37h5ca1d4c_0
-  - termcolor=1.1.0=py_2
-  - text-unidecode=1.3=py_0
-  - threadpoolctl=2.2.0=pyh0d69192_0
-  - tk=8.6.12=h8ffe710_0
-  - toml=0.10.2=pyhd8ed1ab_0
-  - tornado=6.1=py37hcc03f2d_3
-  - tqdm=4.64.0=pyhd8ed1ab_0
-  - typing-extensions=4.2.0=hd8ed1ab_1
-  - typing_extensions=4.2.0=pyha770c72_1
-  - ucrt=10.0.20348.0=h57928b3_0
-  - unicodedata2=14.0.0=py37hcc03f2d_1
-  - unidecode=1.3.4=pyhd8ed1ab_0
-  - urllib3=1.26.9=pyhd8ed1ab_0
-  - vc=14.2=hb210afc_6
-  - vs2015_runtime=14.29.30037=h902a5da_6
-  - waitress=2.1.1=pyhd8ed1ab_0
-  - websocket-client=1.3.2=pyhd8ed1ab_0
-  - werkzeug=2.1.2=pyhd8ed1ab_1
-  - wheel=0.37.1=pyhd8ed1ab_0
-  - win_inet_pton=1.1.0=py37h03978a9_4
-  - wrapt=1.14.1=py37hcc03f2d_0
-  - xorg-libxau=1.0.9=hcd874cb_0
-  - xorg-libxdmcp=1.1.3=hcd874cb_0
-  - xz=5.2.5=h62dcd97_1
-  - yaml=0.2.5=h8ffe710_2
-  - zipp=3.8.0=pyhd8ed1ab_0
-  - zlib=1.2.11=h8ffe710_1014
-  - zstd=1.5.2=h6255e5f_1
+  - docker-py=5.0.3
+  - kaggle=1.5.12
+  - keras-applications=1.0.8
+  - keras-preprocessing=1.1.2
+  - matplotlib=3.5.2
+  - mlflow=1.26.0
+  - numpy=1.21.6
+  - pandas=1.3.5
+  - pip=22.1.1
+  - pymongo=4.1.1
+  - python=3.7.12
+  - sacred=0.8.2
+  - scikit-learn=1.0.2
+  - scipy=1.7.3
+  - seaborn=0.11.2
+  - tensorflow=1.14.0
 prefix: C:\Users\riraa\anaconda3\envs\ium