update handler file

handler.py

@@ -43,10 +43,11 @@ class EndpointHandler:
         # Convert RGB to BGR (PIL uses RGB, OpenCV expects BGR)
         opencv_image = cv2.cvtColor(opencv_image, cv2.COLOR_RGB2BGR)
         output, _ = self.model.enhance(opencv_image, outscale=outscale)
-
+        # BGR to RGB
+        output = cv2.cvtColor(output, cv2.COLOR_BGR2RGB)
+        
         img_byte_arr = BytesIO()
         output = Image.fromarray(output)
-        print(output.size)
         output.save(img_byte_arr, format='PNG')
         img_str = base64.b64encode(img_byte_arr.getvalue())
         return {"out_image": img_str.decode()}