diff --git a/image_prediction/redai_adapter/model_wrapper.py b/image_prediction/redai_adapter/model_wrapper.py
index a90c342..5bb9a70 100644
--- a/image_prediction/redai_adapter/model_wrapper.py
+++ b/image_prediction/redai_adapter/model_wrapper.py
@@ -66,10 +66,6 @@ class EfficientNetWrapper:
 
         model = tf.keras.models.Model(inputs=input_img, outputs=finetuned)
 
-        model.compile(
-            loss="categorical_crossentropy",
-            optimizer="adam",
-            metrics=[tf.keras.metrics.Recall(), tf.keras.metrics.Precision()],
-        )
+        model.compile()
 
         return model