Spaces:

malmukhtar
/

ImageDetection

Runtime error

malmukhtar commited on Apr 16, 2023

Commit

374ef8e

1 Parent(s): a132f61

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -48,8 +48,8 @@ net.load_state_dict(load_url(model_url,map_location=device,check_hash=True))
 transf = utils.get_transformer(face_policy, face_size, net.get_normalizer(), train=False)
 facedet = BlazeFace().to(device)
-facedet.load_weights("../blazeface/blazeface.pth")
-facedet.load_anchors("../blazeface/anchors.npy")
 face_extractor = FaceExtractor(facedet=facedet)
 title = "FaceForensics++"

 transf = utils.get_transformer(face_policy, face_size, net.get_normalizer(), train=False)
 facedet = BlazeFace().to(device)
+facedet.load_weights("./icpr2020dfdc/blazeface/blazeface.pth")
+facedet.load_anchors("./icpr2020dfdc/blazeface/anchors.npy")
 face_extractor = FaceExtractor(facedet=facedet)
 title = "FaceForensics++"