multimodalart HF staff commited on
Commit
d75def4
1 Parent(s): 941bc2f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -51,7 +51,7 @@ async (aspect) => {
51
  def process_sketch(canvas_data, binary_matrixes):
52
  base64_img = canvas_data['image']
53
  image_data = base64.b64decode(base64_img.split(',')[1])
54
- image = Image.open(BytesIO(image_data))
55
  im2arr = np.array(image)
56
  colors = [tuple(map(int, rgb[4:-1].split(','))) for rgb in canvas_data['colors']]
57
  colors_fixed = []
 
51
  def process_sketch(canvas_data, binary_matrixes):
52
  base64_img = canvas_data['image']
53
  image_data = base64.b64decode(base64_img.split(',')[1])
54
+ image = Image.open(BytesIO(image_data)).convert("RGB")
55
  im2arr = np.array(image)
56
  colors = [tuple(map(int, rgb[4:-1].split(','))) for rgb in canvas_data['colors']]
57
  colors_fixed = []