Updated GNN Image

gnn_model/gnn_controller.yaml

@@ -36,7 +36,7 @@ spec:
       nodeName: sonem-worker #The node of the Deployment
       containers:
         - name: gnn-controller
-          image: hecodeco/pdlc-dl-gnns-stgnn-controller:2.0.0-multiarch
+          image: hecodeco/pdlc-dl-gnns-stgnn-controller:2.0.1
           # imagePullPolicy: Always
           env:
           - name: DATA_PATH