cc1234 commited on
Commit
3e3b9fc
·
1 Parent(s): 06fee79

chore: remove unused imports from face_recognition, image_processor, vtt_parser, and interface modules

Browse files
models/face_recognition.py CHANGED
@@ -1,4 +1,3 @@
1
- import os
2
  import numpy as np
3
  from typing import Dict, List, Tuple
4
 
 
 
1
  import numpy as np
2
  from typing import Dict, List, Tuple
3
 
models/image_processor.py CHANGED
@@ -3,10 +3,8 @@ import base64
3
  import numpy as np
4
  from uuid import uuid4
5
  from PIL import Image as PILImage
6
- from typing import List, Dict, Any, Tuple
7
 
8
  from models.face_recognition import EnsembleFaceRecognition, extract_faces, extract_faces_mediapipe
9
- from models.data_manager import DataManager
10
  from utils.vtt_parser import parse_vtt_offsets
11
 
12
  def get_face_predictions(face, ensemble, data_manager, results):
@@ -80,6 +78,7 @@ def image_search_performers(image, data_manager, threshold=0.5, results=3):
80
 
81
  response.append({
82
  'image': im_b64,
 
83
  'confidence': face['confidence'],
84
  'performers': performers
85
  })
 
3
  import numpy as np
4
  from uuid import uuid4
5
  from PIL import Image as PILImage
 
6
 
7
  from models.face_recognition import EnsembleFaceRecognition, extract_faces, extract_faces_mediapipe
 
8
  from utils.vtt_parser import parse_vtt_offsets
9
 
10
  def get_face_predictions(face, ensemble, data_manager, results):
 
78
 
79
  response.append({
80
  'image': im_b64,
81
+ 'area': area,
82
  'confidence': face['confidence'],
83
  'performers': performers
84
  })
utils/vtt_parser.py CHANGED
@@ -1,4 +1,4 @@
1
- from typing import List, Tuple, Generator
2
 
3
  def parse_vtt_offsets(vtt_content: bytes) -> Generator[Tuple[int, int, int, int, float], None, None]:
4
  """
 
1
+ from typing import Tuple, Generator
2
 
3
  def parse_vtt_offsets(vtt_content: bytes) -> Generator[Tuple[int, int, int, int, float], None, None]:
4
  """
web/interface.py CHANGED
@@ -1,5 +1,4 @@
1
  import gradio as gr
2
- from typing import Dict, Any
3
 
4
  from models.data_manager import DataManager
5
  from models.image_processor import (
 
1
  import gradio as gr
 
2
 
3
  from models.data_manager import DataManager
4
  from models.image_processor import (