Update app.py

app.py

@@ -1,7 +1,10 @@
 import gradio as gr
+from datasets import load_dataset
+
+dataset = load_dataset("rebase/hello")
 
 def image_classifier(inp):
-    return {'cat': 0.3, 'dog': 0.7}
+    return dataset
 
 demo = gr.Interface(fn=image_classifier, inputs="image", outputs="label")
 demo.launch()