Update app.py

app.py

@@ -19,13 +19,15 @@ def predict(image):
     try:
         logging.info("Received image of type: %s", type(image))
         logging.debug("Image content: %s", image)
-        # Convert the dictionary to a PIL image
+        
+        # Use the 'composite' key to get the final image
         if isinstance(image, dict):
-            image = image['image']
+            image = image['composite']
+        
         logging.debug("Converting to NumPy array...")
         image = np.array(image).astype('uint8')
         logging.debug("Converting NumPy array to PIL image...")
-        image = Image.fromarray(image, 'RGB')
+        image = Image.fromarray(image, 'RGBA').convert('RGB')
         logging.debug("Image converted successfully.")
 
         logging.info("Processing image...")