arnabg95 commited on
Commit
63d1aab
·
1 Parent(s): a9138a4

package update

Browse files
Files changed (1) hide show
  1. app.py +16 -2
app.py CHANGED
@@ -14,5 +14,19 @@ def greet_json():
14
  return {"Hello": "World!"}
15
 
16
  @app.get("/verify")
17
- def verify(v: Verify):
18
- return v.model_dump()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
14
  return {"Hello": "World!"}
15
 
16
  @app.get("/verify")
17
+ def verify(v: Verify) -> bool:
18
+ data = v.model_dump()
19
+ selfie = data["image"]
20
+ gallery = data["images"]
21
+
22
+ true_count = 0
23
+ for image in gallery:
24
+ try:
25
+ result = DeepFace.verify(img1_path=selfie, img2_path=image)
26
+ if result.get("verified", False):
27
+ true_count += 1
28
+ if true_count >= 2:
29
+ return True
30
+ except Exception as e:
31
+ print(e)
32
+ return False