Update stain_normalization.py

stain_normalization.py

@@ -56,7 +56,7 @@ def normalize_stain(source_wsi_file, output):
         plt.axis("off")
     plt.tight_layout()
     # plt.show()
-    img = plt.savefig('tmp/test.png')
-    image = Image.open('tmp/test.png')
+    img = plt.savefig('images/test.png')
+    image = Image.open('images/test.png')
     numpy_array = np.array(image)
     return numpy_array