Commit ddf7fe59 authored by haven-jeon's avatar haven-jeon
Browse files

fix tensorflow vulnerability

parent 1a36be49

pykospacing/kospacing.py

+1 −1
Original line number Diff line number Diff line
@@ -4,7 +4,7 @@ import re 


import numpy as np

import pkg_resources

from keras.models import load_model

from tensorflow.keras.models import load_model

from pykospacing.embedding_maker import encoding_and_padding, load_vocab



__all__ = ['spacing', ]

requirements.txt

+2 −3
Original line number Diff line number Diff line 
tensorflow == 2.4.0

keras >= 2.4.3

h5py == 2.10.0
 
tensorflow == 2.5.0

h5py == 3.1.0

setup.py

+2 −3
Original line number Diff line number Diff line
@@ -15,9 +15,8 @@ setup(name='pykospacing', 
      include_package_data=True,



      install_requires=[

          'tensorflow == 2.4.0',

          'keras >= 2.4.3',

          'h5py == 2.10.0',

          'tensorflow == 2.5.0',

          'h5py == 3.1.0',

          'argparse >= 1.4.0',

      ],