Update app.py

app.py

@@ -15,17 +15,16 @@ model = VisionEncoderDecoderModel.from_pretrained('microsoft/trocr-large-handwri
 
 def display_sketch(sketch):
     logging.debug(f"Received sketch data: {sketch}")
-    
+
     if isinstance(sketch, dict) and "composite" in sketch:
         image_data = sketch["composite"]
         logging.debug(f"Image data type: {type(image_data)}")
         logging.debug(f"Image data shape: {np.array(image_data).shape}")
-        
+
         plt.imshow(image_data, cmap='gray')
         plt.axis('off')
         
-        # Use a temporary directory for saving the image
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as temp_file:
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".png", dir='/mnt/data') as temp_file:
             plt.savefig(temp_file.name, bbox_inches='tight')
             temp_file_path = temp_file.name
         
@@ -50,7 +49,7 @@ with gr.Blocks() as demo:
     output_image = gr.Image(label="Your Sketch")
     recognized_text = gr.Textbox(label="Recognized Text")
     submit_btn = gr.Button("Submit")
-    
+
     submit_btn.click(fn=display_sketch, inputs=sketchpad, outputs=output_image)
     submit_btn.click(fn=recognize_text, inputs=output_image, outputs=recognized_text)