Update app.py

app.py

@@ -2,7 +2,6 @@ import subprocess
 import sys
 from os.path import abspath, dirname,join
 sys.path.append(join(dirname(abspath(__file__)),'GroundingDINO'))
-print(join(dirname(abspath(__file__)),'GroundingDINO'))
 def run_commands():
     commands = [
         "git clone https://github.com/IDEA-Research/GroundingDINO.git",
@@ -97,7 +96,7 @@ if __name__ == "__main__":
             logit, predicted = torch.max(output.data, 1)
             return self.labels[predicted[0].item()], logit[0].item()
 
-
+    
     class VideoObjectDetection:
 
         def __init__(self,
@@ -167,6 +166,7 @@ if __name__ == "__main__":
             image_transformed, _ = transform(image_pillow, None)
             return image_transformed
 
+        @spaces.GPU(duration=30)
         def generate_video(self, video_path) -> None:
 
             # Load model, set up variables and get video properties
@@ -218,7 +218,6 @@ if __name__ == "__main__":
             frame_count += 1
 
 
-    @spaces.GPU(duration=200)
     def video_object_classification_pipeline():
         video_annotator = VideoObjectDetection(
             text_prompt='human face')
@@ -232,5 +231,4 @@ if __name__ == "__main__":
 
         iface.launch(share=False, debug=True)
 
-    print("Só me falta a GPU")
     video_object_classification_pipeline()