Update handler.py

handler.py

@@ -5,6 +5,7 @@ import base64
 from io import BytesIO
 from PIL import Image
 import os
+import tempfile
 
 class EndpointHandler:
     def __init__(self, model_dir = 'deepseek-ai/DeepSeek-OCR'):
@@ -77,19 +78,22 @@ class EndpointHandler:
             # Define the prompt for Markdown conversion
             prompt = "<image>\n<|grounding|>Convert the document to markdown."
             
-            # Run OCR inference
-            result = self.model.infer(
-                self.tokenizer,
-                prompt=prompt,
-                image_file=image,
-                output_path=None,  # Use temporary directory
-                base_size=1024,
-                image_size=640,
-                crop_mode=True,
-                save_results=False  # Save to temporary directory
-            )
-            
-            return result
+            with tempfile.TemporaryDirectory() as temp_dir:
+                print(f"Using temporary directory: {temp_dir}")
+                
+                    # Run OCR inference
+                result = self.model.infer(
+                    self.tokenizer,
+                    prompt=prompt,
+                    image_file=image,  # Pass the PIL Image object directly
+                    output_path=temp_dir,
+                    base_size=1024,
+                    image_size=640,
+                    crop_mode=True,
+                    save_results=False
+                )
+                    
+                return result
         
         except Exception as e:
             print(f"Error processing image: {e}")