Spaces:
Runtime error
Runtime error
Shafeek Saleem
commited on
Commit
•
34105ec
1
Parent(s):
c5dfc36
bug fixed images
Browse files- utils/database.py +1 -1
- utils/inference.py +1 -0
utils/database.py
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
import os
|
2 |
-
import pickle
|
3 |
|
4 |
def get_database(PKL_PATH):
|
5 |
database = {}
|
|
|
1 |
import os
|
2 |
+
import pickle
|
3 |
|
4 |
def get_database(PKL_PATH):
|
5 |
database = {}
|
utils/inference.py
CHANGED
@@ -9,6 +9,7 @@ PKL_PATH = 'dataset/database.pkl'
|
|
9 |
|
10 |
def recognize(image,tolerance):
|
11 |
database = get_database(PKL_PATH)
|
|
|
12 |
known_encoding = [database[id]['encoding'] for id in database.keys()]
|
13 |
name = 'Unknown'
|
14 |
face_id = 'Unknown'
|
|
|
9 |
|
10 |
def recognize(image,tolerance):
|
11 |
database = get_database(PKL_PATH)
|
12 |
+
st.write(database)
|
13 |
known_encoding = [database[id]['encoding'] for id in database.keys()]
|
14 |
name = 'Unknown'
|
15 |
face_id = 'Unknown'
|