Update app.py

app.py

@@ -1,6 +1,8 @@
 import gradio as gr
 import numpy as np
 import matplotlib.pyplot as plt
+import tempfile
+import os
 import logging
 
 logging.basicConfig(level=logging.DEBUG)
@@ -15,8 +17,13 @@ def display_sketch(sketch):
         
         plt.imshow(image_data, cmap='gray')
         plt.axis('off')
-        plt.savefig("/mnt/data/output.png", bbox_inches='tight')
-        return "/mnt/data/output.png"
+        
+        # Use a temporary directory for saving the image
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as temp_file:
+            plt.savefig(temp_file.name, bbox_inches='tight')
+            temp_file_path = temp_file.name
+        
+        return temp_file_path
     else:
         error_message = f"Unexpected sketch data format: {type(sketch)}"
         logging.error(error_message)