Spaces:
Running
Running
import os | |
# Set DeepFace home directory | |
os.environ["DEEPFACE_HOME"] = "." | |
os.environ["CUDA_VISIBLE_DEVICES"] = "-1" | |
from models.data_manager import DataManager | |
from web.interface import WebInterface | |
def main(): | |
"""Main entry point for the application""" | |
# Initialize data manager | |
data_manager = DataManager( | |
faces_path="data/faces.json", | |
performers_zip="data/persons.zip", | |
facenet_index_path="data/face_facenet.voy", | |
arc_index_path="data/face_arc.voy" | |
) | |
# Initialize and launch web interface | |
web_interface = WebInterface(data_manager, default_threshold=0.5) | |
web_interface.launch(server_name="0.0.0.0", server_port=7860, share=False) | |
if __name__ == "__main__": | |
main() | |