aicloud
/
image_to_image
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

This commit is contained in:
SOULOFCINDER 2023-04-04 13:07:28 +08:00
parent 1dc9e69458
commit f9d9830c75
2 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
ArcaneGAN/app.py
View File

@ -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)

1
ArcaneGAN/requirements.txt
View File

@ -2,3 +2,4 @@ facenet_pytorch
torchvision
torch
gradio
huggingface_hub