ParisNeo commited on
Commit
a1a4fb5
·
1 Parent(s): 25b5d1e
Files changed (1) hide show
  1. app.py +7 -8
app.py CHANGED
@@ -124,15 +124,11 @@ class UI():
124
  with gr.Blocks():
125
  with gr.Row():
126
  with gr.Column():
127
- self.faces_list = gr.Dataframe(
128
- headers=["Face Name"],
129
- datatype=["str"],
130
- label="Faces",
131
- value=[[n] for n in self.known_faces_names]
132
- )
133
  self.btn_reset_faces = gr.Button("clear faces list")
 
134
  self.faces_list_status = gr.Label(label="Status")
135
  self.btn_reset_faces.click(self.clear_faces,[],[self.faces_list_status])
 
136
  with gr.Row():
137
  with gr.Accordion(label="Options", open=False):
138
  self.sld_threshold = gr.Slider(1e-2,10,4e-1,step=1e-2,label="Recognition threshold")
@@ -151,6 +147,9 @@ class UI():
151
 
152
  demo.queue().launch()
153
 
 
 
 
154
  def clear_directory(self, directory_path):
155
  """
156
  Recursively removes all files and subdirectories within the specified directory.
@@ -239,8 +238,8 @@ class UI():
239
  self.known_faces.append(finger_print)
240
  self.known_faces_names.append(file.stem)
241
 
242
- if hasattr(self, "faces_list"):
243
- self.faces_list.update(value=[[n] for n in self.known_faces_names])
244
 
245
  def set_face_name(self, face_name):
246
  self.face_name=face_name
 
124
  with gr.Blocks():
125
  with gr.Row():
126
  with gr.Column():
 
 
 
 
 
 
127
  self.btn_reset_faces = gr.Button("clear faces list")
128
+ self.btn_get_known_faces_list = gr.Button("get known faces list")
129
  self.faces_list_status = gr.Label(label="Status")
130
  self.btn_reset_faces.click(self.clear_faces,[],[self.faces_list_status])
131
+ self.btn_get_known_faces_list.click(self.get_known_faces_list,[],[self.faces_list_status])
132
  with gr.Row():
133
  with gr.Accordion(label="Options", open=False):
134
  self.sld_threshold = gr.Slider(1e-2,10,4e-1,step=1e-2,label="Recognition threshold")
 
147
 
148
  demo.queue().launch()
149
 
150
+ def get_known_faces_list(self):
151
+ return "\n".join(self.known_faces_names)
152
+
153
  def clear_directory(self, directory_path):
154
  """
155
  Recursively removes all files and subdirectories within the specified directory.
 
238
  self.known_faces.append(finger_print)
239
  self.known_faces_names.append(file.stem)
240
 
241
+ # if hasattr(self, "faces_list"):
242
+ # self.faces_list.update(value=[[n] for n in self.known_faces_names])
243
 
244
  def set_face_name(self, face_name):
245
  self.face_name=face_name