update
This commit is contained in:
parent
1dc9e69458
commit
f9d9830c75
|
@ -1,11 +1,16 @@
|
||||||
from facenet_pytorch import MTCNN
|
from facenet_pytorch import MTCNN
|
||||||
from torchvision import transforms
|
from torchvision import transforms
|
||||||
|
from huggingface_hub import hf_hub_download
|
||||||
import torch, PIL
|
import torch, PIL
|
||||||
import gradio as gr
|
import gradio as gr
|
||||||
|
|
||||||
modelarcanev4 = "ArcaneGANv0.4/ArcaneGANv0.4.jit"
|
modelarcanev4 = hf_hub_download(repo_id="akhaliq/ArcaneGANv0.4", filename="ArcaneGANv0.4.jit")
|
||||||
modelarcanev3 = "ArcaneGANv0.3/ArcaneGANv0.3.jit"
|
modelarcanev3 = hf_hub_download(repo_id="akhaliq/ArcaneGANv0.3", filename="ArcaneGANv0.3.jit")
|
||||||
modelarcanev2 = "ArcaneGANv0.2/ArcaneGANv0.2.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)
|
mtcnn = MTCNN(image_size=256, margin=80)
|
||||||
|
|
||||||
|
|
|
@ -2,3 +2,4 @@ facenet_pytorch
|
||||||
torchvision
|
torchvision
|
||||||
torch
|
torch
|
||||||
gradio
|
gradio
|
||||||
|
huggingface_hub
|
Loading…
Reference in New Issue