diff --git a/app/requirements.txt b/app/requirements.txt index ca79cb8..7461e28 100644 --- a/app/requirements.txt +++ b/app/requirements.txt @@ -11,6 +11,7 @@ scikit-image==0.19.2 scikit-learn==1.0.2 scipy==1.7.3 setproctitle==1.2.3 +tensorflow_probability==0.19.0 torchsummary==1.5.1 torch==1.8.2 torchvision==0.9.2 \ No newline at end of file