Update app.py

app.py

@@ -28,7 +28,7 @@ def inference(sample_name):
     outputs = session.run(None, {"input": sample['path'].numpy()})
     predicted_age = (outputs[0]*41.172)+127.329
     # Get the image data from the MetaTensor and convert it to a format PIL can handle
-    image_data = sample['path'].numpy()
+    image_data = sample['path'][0].numpy()
     # Denormalize the image data
     for i in range(3):  # Assuming the image has 3 channels
         image_data[i,:,:] = image_data[i,:,:] * stds[i] + means[i]