Update app.py
Browse files
app.py
CHANGED
|
@@ -133,7 +133,7 @@ def face_recognition(webcan_img):
|
|
| 133 |
|
| 134 |
# Identify the person
|
| 135 |
idx = prediction.argmax(-1)
|
| 136 |
-
person_name = image_paths[idx].split('/').split('.')
|
| 137 |
|
| 138 |
return person_name, similarity_score
|
| 139 |
|
|
|
|
| 133 |
|
| 134 |
# Identify the person
|
| 135 |
idx = prediction.argmax(-1)
|
| 136 |
+
person_name = image_paths[idx].split('/')[-1].split('.')[0]
|
| 137 |
|
| 138 |
return person_name, similarity_score
|
| 139 |
|