Deploying to HF

app.py

@@ -48,7 +48,7 @@ class Explainer:
 
     def convert_fig_to_pil(self, fig):
         fig.canvas.draw()
-        data = np.frombuffer(fig.canvas.tostring_rgb(), dtype=np.uint8)
+        data = np.fromstring(fig.canvas.tostring_rgb(), dtype=np.uint8, sep='')
         data = data.reshape(fig.canvas.get_width_height()[::-1] + (3,))
         return PIL.Image.fromarray(data)