Update inference.py

inference.py

@@ -14,7 +14,6 @@ def model_fn(model_dir, context=None):
 
 def input_fn(input_data, content_type, context=None):
     """Deserialize the input data."""
-    logging.info("Entering input_fn")
     if content_type == 'application/x-image' or content_type == 'application/octet-stream':
         image = Image.open(io.BytesIO(input_data))
         return image
@@ -23,7 +22,6 @@ def input_fn(input_data, content_type, context=None):
 
 def predict_fn(data, model_data, context=None):
     """Apply the model to the input data."""
-    logging.info("Entering predict_fn")
     model, processor, device = model_data
 
     # Preprocess the image
@@ -55,10 +53,7 @@ def predict_fn(data, model_data, context=None):
 
 def output_fn(prediction, accept):
     """Serialize the prediction output."""
-    logging.info("Entering output_fn")
     if accept == 'application/json':
         return json.dumps(prediction), 'application/json'
     else:
-        raise ValueError(f"Unsupported response content type: {accept}")
-
-
+        raise ValueError(f"Unsupported response content type: {accept}")