Update biomap/utils.py

biomap/utils.py

@@ -77,8 +77,8 @@ def transform_to_pil(outputs, alpha=0.3):
     # Transform label by saving it then open it
     label = outputs["linear_preds"][0].numpy()
     # image_label = Image.fromarray(label, mode="P")
-    plt.imsave("output/label.png", label, cmap=cmap)
-    image_label = Image.open("output/label.png")
+    plt.imsave("label.png", label, cmap=cmap)
+    image_label = Image.open("label.png")
     # Overlay labels with img wit alpha
     background = img.convert("RGBA")
     overlay = image_label.convert("RGBA")