update

README.md

@@ -4,7 +4,7 @@ emoji: 💻
 colorFrom: blue
 colorTo: red
 sdk: gradio
-sdk_version: 3.0.5
+sdk_version: 4.13
 app_file: app.py
 pinned: true
 ---

app.py

@@ -28,6 +28,15 @@ gdown.cached_download(
     md5="b1228503d5a89eef7e35e2cbf86b2fc0",
 )
 
+
+gdown.cached_download(
+    url="https://huggingface.co/datasets/taesiri/GamePhysics_Grand_Theft_Auto_V/resolve/main/GTA-V-Videos.zip?download=true",
+    path="./GTA-V-Videos.zip",
+    quiet=False,
+    md5="14d8374a7d1f09c13b5c99fe4d0dc4f1",
+)
+
+
 # EXTRACT
 torchvision.datasets.utils.extract_archive(
     from_path="GTA-V-Embeddings.zip",
@@ -35,13 +44,21 @@ torchvision.datasets.utils.extract_archive(
     remove_finished=False,
 )
 
-# Load videos from Dataset
-gta_v_videos = load_dataset("taesiri/GamePhysics_Grand_Theft_Auto_V")
+
+# EXTRACT
+torchvision.datasets.utils.extract_archive(
+    from_path="GTA-V-Videos.zip",
+    to_path="./",
+    remove_finished=False,
+)
+
+
+list_of_all_videos = glob("GTA-V-Videos/*.mp4")
+
 post_id_to_video_path = {
-    os.path.splitext(os.path.basename(x))[0]: x
-    for x in gta_v_videos["Grand_Theft_Auto_V"][:]["video_file_path"]
+    os.path.splitext(os.path.basename(x))[0]: x for x in list_of_all_videos
 }
-# Initialize CLIP model
+
 clip.available_models()
 
 
@@ -179,22 +196,22 @@ def main():
                 value="A person flying in the air",
                 label="Query",
             ),
-            gr.Radio(list_of_games, label="Game To Search"),
-            gr.Radio(["ViT-B/32"], label="MODEL"),
-            gr.Radio(["Majority", "Top-K"], label="Aggregator"),
+            gr.Radio(list_of_games, label="Game To Search", value=list_of_games[0]),
+            gr.Radio(["ViT-B/32"], label="Model", value="ViT-B/32"),
+            gr.Radio(["Majority", "Top-K"], label="Aggregator", value="Majority"),
             gr.Slider(300, 2000, label="Pool Size", value=1000),
         ],
         outputs=[
-            gr.Textbox(type="auto", label="Search Params"),
-            gr.Video(type="mp4", label="Result 1"),
+            gr.Textbox(label="Search Params"),
+            gr.Video(label="Result 1"),
             gr.Markdown(),
-            gr.Video(type="mp4", label="Result 2"),
+            gr.Video(label="Result 2"),
             gr.Markdown(),
-            gr.Video(type="mp4", label="Result 3"),
+            gr.Video(label="Result 3"),
             gr.Markdown(),
-            gr.Video(type="mp4", label="Result 4"),
+            gr.Video(label="Result 4"),
             gr.Markdown(),
-            gr.Video(type="mp4", label="Result 5"),
+            gr.Video(label="Result 5"),
             gr.Markdown(),
         ],
         examples=[
@@ -244,7 +261,6 @@ def main():
 
     blocks.launch(
         debug=True,
-        enable_queue=True,
     )