fix plot

app.py

@@ -4,6 +4,7 @@ import copy
 from functools import partial
 from io import BytesIO
 import os
+import uuid
 
 from einops import rearrange
 from matplotlib import pyplot as plt
@@ -374,14 +375,15 @@ def make_cluster_plot(eigvecs, images, h=64, w=64, progess_start=0.6, advanced=F
                     i_cluster += 1
         plt.tight_layout(h_pad=0.5, w_pad=0.3)
 
-        buf = BytesIO()
-        plt.savefig(buf, bbox_inches='tight', dpi=72)
+        filename = uuid.uuid4()
+        tmp_path = f"/tmp/{filename}.png"
+        plt.savefig(tmp_path, bbox_inches='tight', dpi=72)
         
-        buf.seek(0)  # Move to the start of the BytesIO buffer
-        img = Image.open(buf)
+        img = Image.open(tmp_path)
         img = img.convert("RGB")
         img = copy.deepcopy(img)
-        buf.close()
+        
+        os.remove(tmp_path)
 
         fig_images.append(img)
         plt.close()
@@ -838,14 +840,17 @@ def plot_one_image_36_grid(original_image, tsne_rgb_images):
             if i_layer == 0:
                 ax.text(-0.1, 0.5, model_name, va="center", ha="center", fontsize=16, transform=ax.transAxes, rotation=90,)    
     plt.tight_layout()
-    buf = BytesIO()
-    plt.savefig(buf, bbox_inches='tight', pad_inches=0, dpi=100)
     
-    buf.seek(0)  # Move to the start of the BytesIO buffer
-    img = Image.open(buf)
+    filename = uuid.uuid4()
+    filename = f"/tmp/{filename}.png"
+    plt.savefig(filename, bbox_inches='tight', pad_inches=0, dpi=100)
+    
+    img = Image.open(filename)
     img = img.convert("RGB")
     img = copy.deepcopy(img)
-    buf.close()
+    
+    os.remove(filename)
+    
     plt.close()
     return img