Spaces:
Running
Running
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 (
|