diff --git a/hpvm/projects/keras/setup.py b/hpvm/projects/keras/setup.py
index bdf192140639569ccaed605ff9238d5ff56a6870..d3a3b295f6cf1ed98027f8e104082ff98856556b 100644
--- a/hpvm/projects/keras/setup.py
+++ b/hpvm/projects/keras/setup.py
@@ -9,10 +9,10 @@ setup(
     author_email='hsharif3@illinois.edu',
     packages=['keras_frontend'],
     install_requires=[
-        "torchvision==0.8.2",
         "tensorflow==1.14",
         "tensorflow-gpu==1.14",
         "keras==2.1.6",
         "scipy==1.1.0"
     ],
+    python_requires="==3.6.*"
 )