update
This commit is contained in:
parent
1dc9e69458
commit
f9d9830c75
|
@ -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)
|
||||
|
||||
|
|
|
@ -2,3 +2,4 @@ facenet_pytorch
|
|||
torchvision
|
||||
torch
|
||||
gradio
|
||||
huggingface_hub
|
Loading…
Reference in New Issue