Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -16,7 +16,7 @@ model, preprocess = clip.load("ViT-B/16", device=device)
|
|
| 16 |
current_model_name = "ViT-B/16"
|
| 17 |
|
| 18 |
# Initialize MTCNN for face detection
|
| 19 |
-
mtcnn = MTCNN(keep_all=False, device=device, thresholds=[0.
|
| 20 |
|
| 21 |
def process_frame(frame, selected_model):
|
| 22 |
global model, preprocess, current_model_name
|
|
|
|
| 16 |
current_model_name = "ViT-B/16"
|
| 17 |
|
| 18 |
# Initialize MTCNN for face detection
|
| 19 |
+
mtcnn = MTCNN(keep_all=False, device=device, thresholds=[0.9, 0.9, 0.9], min_face_size=50)
|
| 20 |
|
| 21 |
def process_frame(frame, selected_model):
|
| 22 |
global model, preprocess, current_model_name
|