diff --git a/ArcaneGAN/app.py b/ArcaneGAN/app.py index c6cb15e..62375d2 100644 --- a/ArcaneGAN/app.py +++ b/ArcaneGAN/app.py @@ -1,11 +1,16 @@ from facenet_pytorch import MTCNN from torchvision import transforms +from huggingface_hub import hf_hub_download import torch, PIL import gradio as gr -modelarcanev4 = "ArcaneGANv0.4/ArcaneGANv0.4.jit" -modelarcanev3 = "ArcaneGANv0.3/ArcaneGANv0.3.jit" -modelarcanev2 = "ArcaneGANv0.2/ArcaneGANv0.2.jit" +modelarcanev4 = hf_hub_download(repo_id="akhaliq/ArcaneGANv0.4", filename="ArcaneGANv0.4.jit") +modelarcanev3 = hf_hub_download(repo_id="akhaliq/ArcaneGANv0.3", filename="ArcaneGANv0.3.jit") +modelarcanev2 = hf_hub_download(repo_id="akhaliq/ArcaneGANv0.2", filename="ArcaneGANv0.2.jit") + +# modelarcanev4 = torch.load("ArcaneGANv0.4/ArcaneGANv0.4.jit") +# modelarcanev3 = torch.load("ArcaneGANv0.3/ArcaneGANv0.3.jit") +# modelarcanev2 = torch.load("ArcaneGANv0.2/ArcaneGANv0.2.jit") mtcnn = MTCNN(image_size=256, margin=80) diff --git a/ArcaneGAN/requirements.txt b/ArcaneGAN/requirements.txt index 0107317..806d5e7 100644 --- a/ArcaneGAN/requirements.txt +++ b/ArcaneGAN/requirements.txt @@ -2,3 +2,4 @@ facenet_pytorch torchvision torch gradio +huggingface_hub \ No newline at end of file