Update app.py

app.py

@@ -18,12 +18,14 @@ model = ViTForImageClassification.from_pretrained(model_name)
 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
         if isinstance(image, dict):
-            logging.debug("Converting dictionary to NumPy array...")
-            image = np.array(image['image_data'])
+            image = image['image']
+        logging.debug("Converting to NumPy array...")
+        image = np.array(image).astype('uint8')
         logging.debug("Converting NumPy array to PIL image...")
-        image = Image.fromarray(image.astype('uint8'), 'RGB')
+        image = Image.fromarray(image, 'RGB')
         logging.debug("Image converted successfully.")
 
         logging.info("Processing image...")