diff --git a/skin_cancer_classification.py b/skin_cancer_classification.py
index d0cacd5..1bd0627 100644
--- a/skin_cancer_classification.py
+++ b/skin_cancer_classification.py
@@ -271,9 +271,8 @@ x = Dense(512, activation='relu')(x) # Added another Dense layer
 x = Dense(256, activation='relu')(x) # Existing Dense layer
 predictions = Dense(1, activation='sigmoid')(x) # Output layer for binary classification
 
-with strategy.scope(): # Use all gpus
-    model = Model(inputs=base_model.input, outputs=predictions)
-    model.compile(optimizer=Adam(learning_rate=0.0001), loss='binary_crossentropy', metrics=['accuracy'])
+model = Model(inputs=base_model.input, outputs=predictions)
+model.compile(optimizer=Adam(learning_rate=0.0001), loss='binary_crossentropy', metrics=['accuracy'])
 
 """## 4. Data Generators