import torch
from src.models.EmojiModel import Generator
from src.models.BgRemover import U2NET
from src.constants import DEVICE


def EmojiFaceGenerator(file_path):
    model=Generator()
    model.load_state_dict(torch.load(file_path,DEVICE))
    return model



def BgRemover(file_path):
    model=U2NET()
    model.load_state_dict(torch.load(file_path,DEVICE))
    return model