test commit

MotuS-Web · Sep 3, 2023 · a6927b1 · a6927b1
1 parent afee5af
commit a6927b1
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 2 deletions.
diff --git a/dummy.mp4 b/dummy.mp4
diff --git a/dummy_skeletons.json b/dummy_skeletons.json
diff --git a/main.py b/main.py
@@ -19,7 +19,7 @@
 EXTRACTOR_THRESHOLD = 0.85
 
 app = FastAPI()
-extractor = SkeletonExtractor(pretrained_bool=True, number_of_keypoints=17, device='mps')
+extractor = SkeletonExtractor(pretrained_bool=True, number_of_keypoints=17, device='cuda')
 preprocessor = DataPreprocessing()
 metrics = Metrics()
 

diff --git a/models.py b/models.py
@@ -227,7 +227,6 @@ def __save_and_read_video_file(self, video, temp_video_file_path):
             for chunk in video.file: f.write(chunk)
         video.file.close()
         video = cv2.VideoCapture(temp_video_file_path)
-        os.remove(temp_video_file_path)
 
         return video