diff --git a/Models/GGSNN-GMN/NeuralNetwork/Dockerfile b/Models/GGSNN-GMN/NeuralNetwork/Dockerfile
index 9915ddb..e77509e 100644
--- a/Models/GGSNN-GMN/NeuralNetwork/Dockerfile
+++ b/Models/GGSNN-GMN/NeuralNetwork/Dockerfile
@@ -12,10 +12,12 @@ RUN pip3 install Cython==0.29.30 && \
     pip3 install pythran==0.11.0 && \
     pip3 install SciPy==1.7.3
 
+RUN pip3 install grpcio==1.48.0
+
 RUN mkdir /code
 COPY requirements.txt /code/
 RUN pip3 install -r /code/requirements.txt
 
 COPY gnn.py /code/
 COPY core/ /code/core/
-COPY model_checkpoint_GGSNN_pair/ /code/model_checkpoint_GGSNN_pair
\ No newline at end of file
+COPY model_checkpoint_GGSNN_pair/ /code/model_checkpoint_GGSNN_pair