Update app.py

app.py

@@ -1,24 +1,24 @@
 import gradio as gr
 
 examples = [
-    "https://ak.picdn.net/shutterstock/videos/21179416/preview/stock-footage-aerial-shot-winter-forest.mp4",
-    "https://ak.picdn.net/shutterstock/videos/5629184/preview/stock-footage-senior-couple-looking-through-binoculars-on-sailboat-together-shot-on-red-epic-for-high-quality-k.mp4",
-    "https://ak.picdn.net/shutterstock/videos/1063125190/preview/stock-footage-a-beautiful-cookie-with-oranges-lies-on-a-green-tablecloth.mp4"
+    ["test1", "https://ak.picdn.net/shutterstock/videos/21179416/preview/stock-footage-aerial-shot-winter-forest.mp4"],
+    ["test2", "https://ak.picdn.net/shutterstock/videos/5629184/preview/stock-footage-senior-couple-looking-through-binoculars-on-sailboat-together-shot-on-red-epic-for-high-quality-k.mp4"],
+    ["test3", "https://ak.picdn.net/shutterstock/videos/1063125190/preview/stock-footage-a-beautiful-cookie-with-oranges-lies-on-a-green-tablecloth.mp4"]
 ]
 
-def predict(video):
-    return "This is a test!", "0.75, 0.25"
+def predict(label, video):
+    return label
 
 with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
-            video = gr.PlayableVideo(label="video")
+            video = gr.PlayableVideo(label="video", title="Input Video")
         with gr.Column():
-            pred = gr.Text(label="prediction")
-            prob = gr.Text(label="probability")
-    btn = gr.Button("Predict")
-    btn.click(predict, inputs=[video], outputs=[pred, prob])
-    gr.Examples(examples=examples, inputs=[video])
+            label = gr.Text(label="ground truth")
+            pred = gr.Text(label="prediction")            
+    btn = gr.Button("Predict", variant="primary"))
+    btn.click(predict, inputs=[label, video], outputs=[pred])
+    gr.Examples(examples=examples, inputs=[label, video], cache_examples=True)
 
 if __name__ == "__main__":
     demo.launch()