Update handler.py

handler.py

@@ -79,7 +79,6 @@ class EndpointHandler:
                 print(f"Image saved to: {image_path}")
                 
                 # Verify the image can be opened
-                output_file = os.path.join(temp_dir, "output-1.md")
                 try:
                     test_image = Image.open(image_path)
                     if test_image.mode != 'RGB':
@@ -94,17 +93,16 @@ class EndpointHandler:
                     self.tokenizer,
                     prompt=prompt,
                     image_file=image_path,  # Pass the PIL Image object directly
-                    output_path=output_file,
+                    output_path=temp_dir,
                     base_size=1024,
                     image_size=640,
                     crop_mode=True,
-                    save_results=True
+                    save_results=False,
+                    eval_mode=True
                 )
-                
-                if os.path.exists(output_file):
-                    with open(output_file, 'r', encoding='utf-8') as f:
-                        content = f.read()
-                    return content
+
+                print(str(result))
+                return result
         
         except Exception as e:
             print(f"Error processing image: {e}")