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 +0 -1
- models/image_processor.py +1 -2
- utils/vtt_parser.py +1 -1
- web/interface.py +0 -1
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
|
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 (
|