diff --git a/ppcls/engine/engine.py b/ppcls/engine/engine.py
index e43b198f5747ef6e52bb2d460f0f12f7d339e438..c24a9163342e381abdc2706d96fe25f4d729e850 100644
--- a/ppcls/engine/engine.py
+++ b/ppcls/engine/engine.py
@@ -394,13 +394,10 @@ class ExportModel(nn.Layer):
         x = self.base_model(x)
         if isinstance(x, list):
             x = x[0]
-
         if self.infer_model_name is not None:
             x = x[self.infer_model_name]
-
         if self.infer_output_key is not None:
             x = x[self.infer_output_key]
-
         if self.softmax is not None:
             x = self.softmax(x)
         return x