Update app.py

app.py

@@ -35,6 +35,10 @@ def process_image(image, k):
     # Convert compressed image back to PIL Image for Gradio output
     compressed_image_pil = Image.fromarray(compressed_image)
     
+    # Ensure the PIL Image is in RGB mode for consistent display
+    if compressed_image_pil.mode != 'RGB':
+        compressed_image_pil = compressed_image_pil.convert('RGB')
+    
     return compressed_image_pil
 
 # Gradio interface
@@ -50,4 +54,4 @@ gr_interface = gr.Interface(
 )
 
 # Launch the Gradio interface
-gr_interface.launch()
+gr_interface.launch()