Update app.py

app.py

@@ -1,7 +1,21 @@
 import gradio as gr
+from datasets import load_dataset
+from itertools import islice
 
-def greet(name):
-    return "Hello " + name + "!!"
+def stream_test(n=50):
+    try:
+        ds = load_dataset("MultimodalUniverse/jwst", split="train", streaming=True)
+        count = sum(1 for _ in islice(ds, int(n)))
+        return f"OK. Streamed {count} JWST records without downloading the full dataset."
+    except Exception as e:
+        return f"Error: {e}"
+
+demo = gr.Interface(
+    fn=stream_test,
+    inputs=gr.Slider(minimum=1, maximum=200, value=50, step=1, label="How many records to stream"),
+    outputs="text",
+    title="JWST streaming check",
+    description="Test that we can stream data from the JWST dataset in small batches."
+)
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
 demo.launch()