Add examples and descriptions

app.py

@@ -11,9 +11,6 @@ from collections import Counter
 
 from SimSearch import FaissCosineNeighbors
 
-# HELPERS
-# to_np = lambda x: x.data.to('cpu').numpy()
-
 # DOWNLOAD THE DATASET and Files
 
 torchvision.datasets.utils.download_file_from_google_drive('1kB1vNdVaNS1OGZ3K8BspBUKkPACCsnrG', '.', 'GTAV-Videos.zip')
@@ -127,8 +124,11 @@ def gradio_search(query, game_name, selected_model, aggregator, pool_size, k=6):
 list_of_games = ['Grand Theft Auto V']
 
 
-
-
+title = "CLIP + GamePhysics - Searching dataset of Gameplay bugs"
+description = "Enter your query and select the game you want to search. The results will be displayed in the console."
+article = """
+This demo shows how to use the CLIP model to search for gameplay bugs in a video game.
+"""
 
 # GRADIO APP
 iface = gr.Interface(fn=gradio_search, 
@@ -136,7 +136,7 @@ iface = gr.Interface(fn=gradio_search,
                                gr.inputs.Radio(list_of_games, label="Game To Search"),
                                gr.inputs.Radio(['ViT-B/32'], label="MODEL"),
                                gr.inputs.Radio(['Majority', 'Top-K'], label="Aggregator"),
-                                gr.inputs.Slider(300, 2000, label="Pool Size"),
+                                gr.inputs.Slider(300, 2000, label="Pool Size", default=1000),
                              ], 
                      outputs=[
                               gr.outputs.Textbox(type="auto", label='Search Params'),
@@ -145,7 +145,19 @@ iface = gr.Interface(fn=gradio_search,
                               gr.outputs.Video(type='mp4', label='Result 3'),
                               gr.outputs.Video(type='mp4', label='Result 4'),
                               gr.outputs.Video(type='mp4', label='Result 5')],
-                     # examples=[],
-                     title='CLIP Meets Game Physics Demo'
+                     examples=[
+                               ["A red car", list_of_games[0], 'ViT-B/32', 'Top-K', 1000],
+                               ["A person wearing pink", list_of_games[0], 'ViT-B/32', 'Top-K', 1000],
+                               ["A car flying in the air", list_of_games[0], 'ViT-B/32', 'Majority', 1000],
+                               ["A person flying in the air",  list_of_games[0], 'ViT-B/32', 'Majority', 1000],
+                               ["A car in vertical position",  list_of_games[0], 'ViT-B/32', 'Majority', 1000],
+                               ["A bike inside a car",  list_of_games[0], 'ViT-B/32', 'Majority', 1000],
+                               ["A bike on a wall",  list_of_games[0], 'ViT-B/32', 'Majority', 1000],
+                               ["A car stuck in a rock",  list_of_games[0], 'ViT-B/32', 'Majority', 1000],
+                               ["A car stuck in a tree",  list_of_games[0], 'ViT-B/32', 'Majority', 1000],
+                               ],
+                     title=title,
+                     description=description,
+                     article=article
                     )
 iface.launch()