Update app.py

app.py

@@ -35,7 +35,7 @@ def plot_vectors(file):
         
         # Save the plot to a buffer
         buf = BytesIO()
-        plt.savefig(buf, format='png')
+        plt.savefig("result.png")
         buf.seek(0)
         plt.close()
         
@@ -44,7 +44,7 @@ def plot_vectors(file):
         img_str = base64.b64encode(buf.getvalue()).decode('utf-8')
         
         # Return the base64-encoded image string
-        return f"data:image/png;base64,{img_str}"
+        return "result.png"
     except Exception as e:
         return str(e)
 
@@ -52,7 +52,7 @@ def plot_vectors(file):
 iface = gr.Interface(
     fn=plot_vectors, 
     inputs=gr.File(label="Upload a .npy file containing a 2D array of 1D vectors"), 
-    outputs=gr.Image(type="pil"), 
+    outputs=gr.Image(type="filepath", label="Visualization Result"), 
     examples=[["random_array.npy"]]  # Example file
 )