Fixing copilot stuff

app.py

@@ -142,7 +142,8 @@ def inference(query, labels, n_supp=10,
         # it used to be a matplotlib figure, turning it into a PIL image
         fig.canvas.draw()
         w, h = fig.canvas.get_width_height()
-        img = np.frombuffer(fig.canvas.tostring_argb(), dtype=np.uint8).reshape(h, w, 3)
+        img = np.frombuffer(fig.canvas.tostring_argb(), dtype=np.uint8).reshape(h, w, 4)
+        img = img[:, :, :3]
         return {k: float(v) for k, v in zip(labels, probs[0, 0])}, Image.fromarray(img)
         #return {k: float(v) for k, v in zip(labels, probs[0, 0])}, fig