updated interface

app.py

@@ -1,8 +1,14 @@
+# Setting global vars 
+CSV_PATH = ""
+VID_PATH = ""
+BATCH_SIZE = 5 # CHANGE THIS TO FIT YOUR GPU, YOU MAY NEED SOME TRIAL AND ERROR
+CUDA_PATH = "/usr/local/cuda-11.8/" # CHANGE THIS TO YOUR LOCAL CUDA PATH
+
 import subprocess
 import os 
 import sys
 # set CUDA_HOME
-os.environ["CUDA_HOME"] = "/usr/local/cuda-11.8/"
+os.environ["CUDA_HOME"] = CUDA_PATH
 os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'max_split_size_mb:5000'
 subprocess.run(['pip', 'install', '-e', 'GroundingDINO'])
 sys.path.append(os.path.join(os.getcwd(), "GroundingDINO"))
@@ -11,8 +17,7 @@ sys.path.append(os.path.join(os.getcwd(), "segment_anything"))
 import gradio as gr
 from dino_sam import sam_dino_vid
 
-CSV_PATH = ""
-VID_PATH = ""
+
 
 def run_sam_dino(input_vid,
                  grounding_caption,
@@ -22,7 +27,7 @@ def run_sam_dino(input_vid,
                  video_options):
     new_input_vid = input_vid.replace(" ", "_")
     os.rename(input_vid, new_input_vid)
-    csv_path, vid_path = sam_dino_vid(new_input_vid, grounding_caption, box_threshold, text_threshold, fps_processed, video_options)
+    csv_path, vid_path = sam_dino_vid(new_input_vid, grounding_caption, box_threshold, text_threshold, fps_processed, video_options, batch_size=BATCH_SIZE)
     global CSV_PATH
     CSV_PATH = csv_path
     global VID_PATH
@@ -46,7 +51,7 @@ with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
             input = gr.Video(label="Input Video", interactive=True)
-            grounding_caption = gr.Textbox(label="What do you want to detect?")
+            grounding_caption = gr.Textbox(label="What do you want to detect? (Muliple species should be separated by commas. i.e. 'baboons, chimpanzees')")
             with gr.Accordion("Advanced Options", open=False):
                 box_threshold = gr.Slider(
                     label="Box Threshold",
@@ -90,5 +95,22 @@ with gr.Blocks() as demo:
         cache_examples=True,
         label='Example'
       )
+    
+    gr.DuplicateButton()
+    
+    gr.Markdown(
+        """
+        ## Frequently Asked Questions 
+        
+        ##### How can I run the interface on my own computer? 
+        By clicking on the three dots on the top right corner of the interface, you will be able to clone the repository or run it with a Docker image on your local machine. \
+        For local machine setup instructions please check the README file. 
+
+        ##### The video is very slow to process, how can I speed it up?
+        You can speed up the processing by adjusting the frame detection rate in the advanced options. The lower the number the slower the processing time. Choosing only\
+        bounding boxes will make the processing faster. You can also duplicate the space using the Duplicate Button and choose a different GPU which will make the processing faster.
+
+        """
+    )
 
 demo.launch(share=False)