Shafeek Saleem commited on
Commit
f16075a
·
1 Parent(s): 0365ce2

bug fixed images

Browse files
Files changed (2) hide show
  1. pages/4_Trying It Out.py +0 -1
  2. 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'