Update app.py

app.py

@@ -6,8 +6,8 @@ description = "Keras implementation for Video Vision Transformer trained with Or
 article = "Classes: liver, kidney-right, kidney-left, femur-right, femur-left, bladder, heart, lung-right, lung-left, spleen, pancreas.\n\nAuthor:<a href=\"https://huggingface.co/pablorodriper/\"> Pablo Rodríguez</a>; Based on the keras example by <a href=\"https://keras.io/examples/vision/vivit/\">Aritra Roy Gosthipaty and Ayush Thakur</a>"
 title = "Video Vision Transformer on OrganMNIST3D"
 
-def infer(x):
-    return model.predict(tf.expand_dims(x, axis=0))[0]
+def infer(video):
+    return model.predict(tf.expand_dims(video, axis=0))[0]
 
 model = from_pretrained_keras("keras-io/video-vision-transformer")
 
@@ -19,7 +19,8 @@ iface = gr.Interface(
     outputs = "number",
     description = description,
     title = title,
-    article = article
+    article = article,
+    examples=["example_1.mp4", "example_2.mp4"]
     )
 
 iface.launch()