init_test

app.py

@@ -273,7 +273,7 @@ def extract_face_from_fixed_num_frames(src_video, dst_path, num_frames=None, dev
 
     video_capture.release()
     # cv2.destroyAllWindows()
-    return frame_index
+    return frame_indices
 
 
 def FSFM3C_video_detection(video):
@@ -287,7 +287,7 @@ def FSFM3C_video_detection(video):
     frame_path = os.path.join(FRAME_SAVE_PATH, str(num_files))
     os.makedirs(frame_path, exist_ok=True)
     os.makedirs(os.path.join(frame_path, '0'), exist_ok=True)
-    frame_index = extract_face_from_fixed_num_frames(video, frame_path, num_frames=num_frames, device=device)
+    frame_indices = extract_face_from_fixed_num_frames(video, frame_path, num_frames=num_frames, device=device)
 
     args.data_path = frame_path
     args.batch_size = 32
@@ -304,7 +304,7 @@ def FSFM3C_video_detection(video):
     frame_preds_list, video_pred_list = test_all(data_loader_val, model, device)
 
     real_prob_frames = [round(1. - fake_score, 2)  for fake_score in video_pred_list]
-    frame_results = {f"frame_{frame}": f"{int(real_prob_frames[i] * 100)}%" for i, frame in enumerate(frame_index)}
+    frame_results = {f"frame_{frame}": f"{int(real_prob_frames[i] * 100)}%" for i, frame in enumerate(frame_indices)}
 
     real_prob_video = int(round(1. - (sum(video_pred_list) / len(video_pred_list)), 2) * 100)
     if real_prob_video > 50:
@@ -372,7 +372,7 @@ with gr.Blocks() as demo:
 
     gr.Markdown("- <b>V1.0 [2024-12] (Current):</b> "
                 "Create this page with basic detectors (simply fine-tuned models) that follow the paper implementation. "
-                "<b>Notes:</b> Performance is limited because no any optimization of data, models, hyperparameters, etc. is done for downstream tasks. \ "
+                "<b>Notes:</b> Performance is limited because no any optimization of data, models, hyperparameters, etc. is done for downstream tasks. <br> "
                 "<b>[TODO]: </b> Update practical models, and optimized interfaces, and provide more functions such as visualizations, a unified detector, and multi-modal diagnosis.")
 
     gr.Markdown(