Update handler.py

handler.py

@@ -88,12 +88,15 @@ class EndpointHandler:
                 except Exception as img_error:
                     return {"error": f"Invalid image: {str(img_error)}"}
 
+                output_dir = os.path.join(temp_dir, "deepseek_out")
+                os.makedirs(output_dir, exist_ok=True)
+                
                  # Run OCR inference
                 result = self.model.infer(
                     self.tokenizer,
                     prompt=prompt,
                     image_file=image_path,  # Pass the PIL Image object directly
-                    output_path=temp_dir,
+                    output_path=output_dir,
                     base_size=1024,
                     image_size=640,
                     crop_mode=True,
@@ -101,9 +104,10 @@ class EndpointHandler:
                     eval_mode=True
                 )
 
-                for fname in os.listdir(temp_dir):
+                for fname in os.listdir(output_dir):
+                    print("File:\n", fname)
                     if fname.endswith(".md") or fname.endswith(".mmd"):
-                        md_path = os.path.join(temp_dir, fname)
+                        md_path = os.path.join(output_dir, fname)
                         with open(md_path, 'r', encoding='utf-8') as f:
                             markdown = f.read()
                         print("Markdown output:\n", markdown)