reab5555 commited on
Commit
9f57102
·
verified ·
1 Parent(s): 07bcae5

Update video_processing.py

Browse files
Files changed (1) hide show
  1. video_processing.py +0 -1
video_processing.py CHANGED
@@ -17,7 +17,6 @@ from facenet_pytorch import MTCNN
17
  import torch
18
  import mediapipe as mp
19
 
20
- @spaces.GPU(duration=300)
21
 
22
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
23
  mtcnn = MTCNN(keep_all=False, device=device, thresholds=[0.95, 0.95, 0.95], min_face_size=80)
 
17
  import torch
18
  import mediapipe as mp
19
 
 
20
 
21
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
22
  mtcnn = MTCNN(keep_all=False, device=device, thresholds=[0.95, 0.95, 0.95], min_face_size=80)