Spaces:
Runtime error
Runtime error
Shafeek Saleem
commited on
Commit
·
f16075a
1
Parent(s):
0365ce2
bug fixed images
Browse files- pages/4_Trying It Out.py +0 -1
- utils/inference.py +3 -1
pages/4_Trying It Out.py
CHANGED
@@ -2,7 +2,6 @@ import streamlit as st
|
|
2 |
from utils.levels import complete_level, render_page, initialize_level
|
3 |
from utils.login import get_login, initialize_login
|
4 |
from utils.inference import recognize
|
5 |
-
from utils.database import get_database
|
6 |
import os
|
7 |
import time
|
8 |
import face_recognition
|
|
|
2 |
from utils.levels import complete_level, render_page, initialize_level
|
3 |
from utils.login import get_login, initialize_login
|
4 |
from utils.inference import recognize
|
|
|
5 |
import os
|
6 |
import time
|
7 |
import face_recognition
|
utils/inference.py
CHANGED
@@ -5,8 +5,10 @@ import cv2
|
|
5 |
import numpy as np
|
6 |
from PIL import Image
|
7 |
|
|
|
|
|
8 |
def recognize(image,tolerance):
|
9 |
-
database = get_databse()
|
10 |
known_encoding = [database[id]['encoding'] for id in database.keys()]
|
11 |
name = 'Unknown'
|
12 |
face_id = 'Unknown'
|
|
|
5 |
import numpy as np
|
6 |
from PIL import Image
|
7 |
|
8 |
+
PKL_PATH = 'dataset/database.pkl'
|
9 |
+
|
10 |
def recognize(image,tolerance):
|
11 |
+
database = get_databse(PKL_PATH)
|
12 |
known_encoding = [database[id]['encoding'] for id in database.keys()]
|
13 |
name = 'Unknown'
|
14 |
face_id = 'Unknown'
|