Update inference.py

inference.py

@@ -1,9 +1,7 @@
-import os
 import torch
 import re
 from PIL import Image
 from transformers import DonutProcessor, VisionEncoderDecoderModel
-import base64
 import io
 import json
 
@@ -17,9 +15,7 @@ def model_fn(model_dir, context=None):
 def transform_fn(model, request_body, request_content_type, response_content_type, context=None):
     model, processor, device = model
     if request_content_type == 'application/json':
-        data = json.loads(request_body)
-        image_data = data['inputs']
-        image = Image.open(io.BytesIO(base64.b64decode(image_data))).convert("RGB")
+        image = Image.open(io.BytesIO(request_body))
         
         # Preprocess the image
         pixel_values = processor(image, return_tensors="pt").pixel_values.to(device)