reab5555 commited on
Commit
f7bab76
·
verified ·
1 Parent(s): 202c430

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -24,7 +24,7 @@ def process_and_show_completion(video_input_path, anomaly_threshold_input, fps,
24
  mse_heatmap_embeddings, mse_heatmap_posture, mse_heatmap_voice, \
25
  face_samples_frequent, \
26
  anomaly_faces_embeddings, anomaly_frames_posture_images, \
27
- aligned_faces_folder, frames_folder, \
28
  stacked_heatmap = results
29
 
30
  anomaly_faces_embeddings_pil = [Image.fromarray(face) for face in anomaly_faces_embeddings] if anomaly_faces_embeddings is not None else []
@@ -40,7 +40,7 @@ def process_and_show_completion(video_input_path, anomaly_threshold_input, fps,
40
  mse_heatmap_embeddings, mse_heatmap_posture, mse_heatmap_voice,
41
  anomaly_faces_embeddings_pil, anomaly_frames_posture_pil,
42
  face_samples_frequent,
43
- aligned_faces_folder, frames_folder,
44
  mse_embeddings, mse_posture, mse_voice,
45
  stacked_heatmap
46
 
@@ -202,7 +202,7 @@ with gr.Blocks() as iface:
202
  mse_features_store = gr.State()
203
  mse_posture_store = gr.State()
204
  mse_voice_store = gr.State()
205
- aligned_faces_folder_store = gr.State()
206
  frames_folder_store = gr.State()
207
  mse_heatmap_embeddings_store = gr.State()
208
  mse_heatmap_posture_store = gr.State()
@@ -219,7 +219,7 @@ with gr.Blocks() as iface:
219
  mse_features_heatmap, mse_posture_heatmap, mse_voice_heatmap,
220
  anomaly_frames_features, anomaly_frames_posture,
221
  face_samples_most_frequent,
222
- aligned_faces_folder_store, frames_folder_store,
223
  mse_heatmap_embeddings_store, mse_heatmap_posture_store, mse_heatmap_voice_store,
224
  stacked_heatmap_plot
225
  ]
 
24
  mse_heatmap_embeddings, mse_heatmap_posture, mse_heatmap_voice, \
25
  face_samples_frequent, \
26
  anomaly_faces_embeddings, anomaly_frames_posture_images, \
27
+ faces_folder, frames_folder, \
28
  stacked_heatmap = results
29
 
30
  anomaly_faces_embeddings_pil = [Image.fromarray(face) for face in anomaly_faces_embeddings] if anomaly_faces_embeddings is not None else []
 
40
  mse_heatmap_embeddings, mse_heatmap_posture, mse_heatmap_voice,
41
  anomaly_faces_embeddings_pil, anomaly_frames_posture_pil,
42
  face_samples_frequent,
43
+ faces_folder, frames_folder,
44
  mse_embeddings, mse_posture, mse_voice,
45
  stacked_heatmap
46
 
 
202
  mse_features_store = gr.State()
203
  mse_posture_store = gr.State()
204
  mse_voice_store = gr.State()
205
+ faces_folder_store = gr.State()
206
  frames_folder_store = gr.State()
207
  mse_heatmap_embeddings_store = gr.State()
208
  mse_heatmap_posture_store = gr.State()
 
219
  mse_features_heatmap, mse_posture_heatmap, mse_voice_heatmap,
220
  anomaly_frames_features, anomaly_frames_posture,
221
  face_samples_most_frequent,
222
+ faces_folder_store, frames_folder_store,
223
  mse_heatmap_embeddings_store, mse_heatmap_posture_store, mse_heatmap_voice_store,
224
  stacked_heatmap_plot
225
  ]