Spaces:

WildOjisan
/

insightface_hf_2509

Sleeping

WildOjisan commited on Feb 20

Commit

e2353df

1 Parent(s): f362a25

.

Files changed (1) hide show

routers/predict.py CHANGED Viewed

@@ -51,7 +51,7 @@ def extract_adaface_embedding(img_bgr, face_kps):
         embedding, _ = adaface_model(img_tensor)
     return embedding.cpu().numpy()[0]
-@router.post("/")
 async def predict(file: UploadFile = File(...)):
     contents = await file.read()
     nparr = np.frombuffer(contents, np.uint8)

         embedding, _ = adaface_model(img_tensor)
     return embedding.cpu().numpy()[0]
+@router.post("/predict")
 async def predict(file: UploadFile = File(...)):
     contents = await file.read()
     nparr = np.frombuffer(contents, np.uint8)