Update app.py

app.py

@@ -21,11 +21,11 @@ def display_sketch(sketch):
         logging.debug(f"Image data type: {type(image_data)}")
         logging.debug(f"Image data shape: {image_data.shape}")
 
-        plt.imshow(image_data)
-        plt.axis('off')
+        # Ensure the image is in the correct format
+        image = Image.fromarray(image_data, 'RGBA').convert('RGB')
 
         temp_file_path = os.path.join(os.getcwd(), "output.png")
-        plt.savefig(temp_file_path, bbox_inches='tight', pad_inches=0)
+        image.save(temp_file_path)
         logging.debug(f"Image saved to: {temp_file_path}")
 
         return temp_file_path
@@ -43,6 +43,7 @@ def recognize_text(image_path):
         # Generate the text
         generated_ids = model.generate(pixel_values)
         generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
+        logging.debug(f"Recognized text: {generated_text}")
         return generated_text
     except Exception as e:
         logging.error(f"Error in recognizing text: {e}")