Update app.py

app.py

@@ -3,14 +3,15 @@ import matplotlib.pyplot as plt
 import numpy as np
 from datasets import load_dataset
 
-# Load the chest X-ray classification dataset
-dataset = load_dataset("keremberke/chest-xray-classification")
+# Load the chest X-ray classification dataset with the 'full' configuration
+dataset = load_dataset("keremberke/chest-xray-classification", "full")
 
 def show_samples(label):
     # Get samples from the dataset
     images = []
     for i in range(5):  # Show 5 images
-        images.append(dataset['train'][i]['image'])  # Adjust as needed
+        img = dataset['train'][i]['image']  # Adjust as needed
+        images.append(img)
 
     # Create a grid of images
     fig, axes = plt.subplots(1, 5, figsize=(15, 5))