Update app/gradio_app.py

app/gradio_app.py

@@ -30,9 +30,10 @@ def classify_image(image):
         return image, {"error": f"Classification failed: {str(e)}"}
 
 def get_random_image():
-    # Pick a random sample from the dataset
-    sample = ds.shuffle(seed=random.randint(0, 9999))[0]
-    img = sample["image"].convert("RGB")  # just ensure RGB mode
+    # Pick a random index from the entire dataset
+    idx = random.randint(0, len(ds) - 1)
+    sample = ds[idx]
+    img = sample["image"].convert("RGB")  # ensure RGB mode
     return img
 
 with gr.Blocks(title="Cats vs Dogs Classifier", theme=gr.themes.Soft()) as demo: