Update pred_color.py

pred_color.py

@@ -70,8 +70,8 @@ def conditioning_from_landmarks(landmarks, size=512):
 
     fig.canvas.draw()
     buffer, (width, height) = fig.canvas.print_to_buffer()
-    assert width == height
-    assert width == size
+    #assert width == height
+    #assert width == size
 
     buffer = np.frombuffer(buffer, np.uint8).reshape((height, width, 4))
     buffer = buffer[:, :, 0:3]