From f9d9830c75a8553f106ebb06b972794f9e8e46ac Mon Sep 17 00:00:00 2001
From: SOULOFCINDER <841135647@qq.com>
Date: Tue, 4 Apr 2023 13:07:28 +0800
Subject: [PATCH] update

---
 ArcaneGAN/app.py           | 11 ++++++++---
 ArcaneGAN/requirements.txt |  1 +
 2 files changed, 9 insertions(+), 3 deletions(-)

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