promptTrack installed

app.py

@@ -159,12 +159,9 @@ import gradio as gr
 import shutil
 import os
 import subprocess
-import gradio as gr
-import shutil
-import os
 import sys
 # Run the .bat file before launching the app
-try:
+"""try:
     import PromptTrack
 except ImportError:
     print("PromptTrack not found. Installing...")
@@ -178,10 +175,7 @@ except ImportError:
 
 
 from PromptTrack import PromptTracker
-
-#initialize PromptTrack
-
-tracker = PromptTracker()
+tracker = PromptTracker()"""
 def process_video(video_path, prompt):
     detection_threshold=0.3
     track_thresh=0.4
@@ -192,8 +186,8 @@ def process_video(video_path, prompt):
     output_file = video_path.split('mp4')[0]+"_mot_.json"    # Tracking result
     output_file_2 = video_path.split('mp4')[0]+"_object_detection.json"    # detection results
 
-    """video_file = video_path
-    tracker.detect_objects(video_file, prompt=prompt, nms_threshold=0.8, detection_threshold=detection_threshold, detector="OWL-VITV2")
+    video_file = video_path
+    """tracker.detect_objects(video_file, prompt=prompt, nms_threshold=0.8, detection_threshold=detection_threshold, detector="OWL-VITV2")
     tracker.process_mot(video_file, fixed_parc=True, track_thresh=track_thresh, match_thresh=match_thresh, frame_rate=25, max_time_lost=max_time_lost, nbr_frames_fixing=nbr_frames_fixing)
     tracker.read_video_with_mot(video_file, fps=25)
     """

requirements.txt

@@ -1,7 +1,7 @@
 
-# accelerate
-# diffusers
-# invisible_watermark
-# torch
-# transformers
-# xformers
+accelerate
+diffusers
+invisible_watermark
+torch
+transformers
+xformers