spaces

app.py

@@ -6,8 +6,8 @@ import contextlib
 import io
 import traceback
 import gradio as gr
+import spaces
 from transformers import AutoModel, AutoTokenizer, AutoConfig, pipeline
-
 from huggingface_hub import snapshot_download
 import sys, pathlib
 import os
@@ -52,6 +52,7 @@ gd_config_path = "//" + str(current_dir / "GroundingDINO_SwinT_OGC.py")
 gd_checkpoint_path = "//" + str(current_dir / "groundingdino_swint_ogc.pth")
 visualization_dir = "//" + str(current_dir / "outputs")
 
+@spaces.GPU(duration=200)
 def process_video(video_file, categorical_keywords, unary_keywords, binary_keywords, object_pairs, output_fps):
     log_buffer = io.StringIO()
     try:
@@ -143,4 +144,12 @@ demo = gr.Interface(
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()
+    
+    
+    
+        
+
+
+#input would be video file path and keywords
+#out would be video, efforts