Initial Commit

README.md

@@ -1,12 +1,11 @@
 ---
-title: Winoground Demo
-emoji: 📊
-colorFrom: green
-colorTo: purple
+title: WinoGround Demo
+emoji: 💻
+colorFrom: blue
+colorTo: gray
 sdk: gradio
-sdk_version: 3.0.19
 app_file: app.py
 pinned: false
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces#reference

app.py

@@ -0,0 +1,40 @@
+from datasets import load_dataset
+import gradio as gr
+
+
+winoground = None
+
+def fetch(auth_token):
+    global winoground
+    if winoground is None:
+        winoground = load_dataset("facebook/winoground", use_auth_token=auth_token)["test"]
+    example = winoground[0]
+    return example["image_0"], example["caption_0"], example["image_1"], example["caption_1"]
+
+def func(index):
+    example = winoground[index]
+    return example["image_0"], example["caption_0"], example["image_1"], example["caption_1"]
+
+
+demo = gr.Blocks()
+
+
+with demo:
+    gr.Markdown("Slide across the slider to see various examples from WinoGround")
+
+    with gr.Column():
+        auth_token = gr.Textbox(label="Hugging Face Access Token")
+        button = gr.Button("Get dataset")
+        slider = gr.Slider(minimum=0, maximum=400)
+        with gr.Row():
+            with gr.Column():
+                image_input_1 = gr.Image()
+                text_input_1 = gr.Textbox()
+            with gr.Column():
+                image_input_2 = gr.Image()
+                text_input_2 = gr.Textbox()
+    
+    button.click(fetch, inputs=[auth_token], outputs=[image_input_1, text_input_1, image_input_2, text_input_2])
+    slider.change(func, inputs=[slider], outputs=[image_input_1, text_input_1, image_input_2, text_input_2])
+
+demo.launch()

requirements.txt

@@ -0,0 +1,2 @@
+datasets
+gradio