Update app.py

app.py

@@ -16,15 +16,15 @@ model = VisionEncoderDecoderModel.from_pretrained('microsoft/trocr-large-handwri
 def display_sketch(sketch):
     logging.debug(f"Received sketch data: {sketch}")
 
-    if isinstance(sketch, np.ndarray):
-        image_data = sketch
+    if isinstance(sketch, dict) and 'composite' in sketch:
+        image_data = np.array(sketch['composite'], dtype=np.uint8)
         logging.debug(f"Image data type: {type(image_data)}")
-        logging.debug(f"Image data shape: {np.array(image_data).shape}")
+        logging.debug(f"Image data shape: {image_data.shape}")
 
-        plt.imshow(image_data, cmap='gray')
+        plt.imshow(image_data)
         plt.axis('off')
         
-        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', pad_inches=0)
             temp_file_path = temp_file.name