[saliency.view] Properly get image from sample data

src/mednet/classify/engine/saliency/viewer.py

@@ -199,7 +199,7 @@ def run(
             # not the case.
             name = str(sample[1]["name"][0])
             label = int(sample[1]["target"].item())
-            data = sample[0][0]
+            data = sample[0]["image"][0]
 
             if label != target_label:
                 # no visualisation was generated