Spaces:
Sleeping
Sleeping
Update detection.py
Browse files- detection.py +1 -1
detection.py
CHANGED
@@ -27,7 +27,7 @@ face_app.prepare(ctx_id=0) # Prepare for GPU inference
|
|
27 |
reid_extractor = torchreid.utils.FeatureExtractor(
|
28 |
model_name='osnet_x0_25',
|
29 |
model_path='osnet_x0_25_market1501.pth', # Pre-trained on Market1501 dataset
|
30 |
-
device='cuda'
|
31 |
)
|
32 |
|
33 |
# =============================================================================
|
|
|
27 |
reid_extractor = torchreid.utils.FeatureExtractor(
|
28 |
model_name='osnet_x0_25',
|
29 |
model_path='osnet_x0_25_market1501.pth', # Pre-trained on Market1501 dataset
|
30 |
+
device='cuda'
|
31 |
)
|
32 |
|
33 |
# =============================================================================
|