diff --git a/kerasDeloyment.yaml b/kerasDeloyment.yaml
index cb88811bfd3e6ccc4c0ed6feaa3dfb675bc8cf0a..bc8890083ea0e35ce1da2801c016357b755343db 100644
--- a/kerasDeloyment.yaml
+++ b/kerasDeloyment.yaml
@@ -44,12 +44,12 @@ spec:
           persistentVolumeClaim:
             claimName: modeltraining
             
-    #   affinity:
-    #     nodeAffinity:
-    #       requiredDuringSchedulingIgnoredDuringExecution:
-    #         nodeSelectorTerms:
-    #         - matchExpressions:
-    #           - key: gpu-type
-    #             operator: In # Use NotIn for other types
-    #             values:
-    #             - V100
+      affinity:
+        nodeAffinity:
+          requiredDuringSchedulingIgnoredDuringExecution:
+            nodeSelectorTerms:
+            - matchExpressions:
+              - key: gpu-type
+                operator: In # Use NotIn for other types
+                values:
+                - V100