MAINT: put back models due to 403 in HF Hub

app.py

@@ -1,5 +1,3 @@
-from cgitb import enable
-from ctypes.wintypes import HFONT
 import os
 import sys
 import torch
@@ -10,7 +8,6 @@ import torchvision.transforms as transforms
 
 from torch.autograd import Variable
 from network.Transformer import Transformer
-from huggingface_hub import hf_hub_download
 
 from PIL import Image
 
@@ -19,8 +16,6 @@ import logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# Constants
-
 MAX_DIMENSION = 1280
 MODEL_PATH = "models"
 COLOUR_MODEL = "RGB"
@@ -32,37 +27,23 @@ STYLE_KON = "Satoshi Kon"
 DEFAULT_STYLE = STYLE_SHINKAI
 STYLE_CHOICE_LIST = [STYLE_SHINKAI, STYLE_HOSODA, STYLE_MIYAZAKI, STYLE_KON]
 
-MODEL_REPO_ID = "akiyamasho/AnimeBackgroundGAN"
-MODEL_FILE_SHINKAI = "shinkai_makoto.pth"
-MODEL_FILE_HOSODA = "hosoda_mamoru.pth"
-MODEL_FILE_MIYAZAKI = "miyazaki_hayao.pth"
-MODEL_FILE_KON = "kon_satoshi.pth"
-
-# Model Initalisation
-shinkai_model_hfhub = hf_hub_download(repo_id=MODEL_REPO_ID, filename=MODEL_FILE_SHINKAI)
-hosoda_model_hfhub = hf_hub_download(repo_id=MODEL_REPO_ID, filename=MODEL_FILE_HOSODA)
-miyazaki_model_hfhub = hf_hub_download(repo_id=MODEL_REPO_ID, filename=MODEL_FILE_MIYAZAKI)
-kon_model_hfhub = hf_hub_download(repo_id=MODEL_REPO_ID, filename=MODEL_FILE_KON)
-
 shinkai_model = Transformer()
 hosoda_model = Transformer()
 miyazaki_model = Transformer()
 kon_model = Transformer()
 
-enable_gpu = torch.cuda.is_available()
-map_location = torch.device("cuda") if enable_gpu else "cpu"
 
 shinkai_model.load_state_dict(
-    torch.load(shinkai_model_hfhub, map_location=map_location)
+    torch.load(os.path.join(MODEL_PATH, "shinkai_makoto.pth"))
 )
 hosoda_model.load_state_dict(
-    torch.load(hosoda_model_hfhub, map_location=map_location)
+    torch.load(os.path.join(MODEL_PATH, "hosoda_mamoru.pth"))
 )
 miyazaki_model.load_state_dict(
-    torch.load(miyazaki_model_hfhub, map_location=map_location)
+    torch.load(os.path.join(MODEL_PATH, "miyazaki_hayao.pth"))
 )
 kon_model.load_state_dict(
-    torch.load(kon_model_hfhub, map_location=map_location)
+    torch.load(os.path.join(MODEL_PATH, "kon_satoshi.pth"))
 )
 
 shinkai_model.eval()
@@ -70,7 +51,8 @@ hosoda_model.eval()
 miyazaki_model.eval()
 kon_model.eval()
 
-# Functions
+enable_gpu = torch.cuda.is_available()
+
 
 def get_model(style):
     if style == STYLE_SHINKAI:
@@ -127,8 +109,6 @@ def inference(img, style):
     return transforms.ToPILImage()(output_image)
 
 
-# Gradio setup
-
 title = "Anime Background GAN"
 description = "Gradio Demo for CartoonGAN by Chen Et. Al. Models are Shinkai Makoto, Hosoda Mamoru, Kon Satoshi, and Miyazaki Hayao."
 article = "<p style='text-align: center'><a href='http://openaccess.thecvf.com/content_cvpr_2018/CameraReady/2205.pdf' target='_blank'>CartoonGAN Whitepaper from Chen et.al</a></p><p style='text-align: center'><a href='https://github.com/venture-anime/cartoongan-pytorch' target='_blank'>Github Repo</a></p><p style='text-align: center'><a href='https://github.com/Yijunmaverick/CartoonGAN-Test-Pytorch-Torch' target='_blank'>Original Implementation from Yijunmaverick</a></p><center><img src='https://visitor-badge.glitch.me/badge?page_id=akiyamasho' alt='visitor badge'></center></p>"

models/hosoda_mamoru.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c666eea7700864d5972765cc43e926d900174648297bfef494006dc230fd1bf0
+size 44529096

models/kon_satoshi.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0629352a54838e56a2ad7fca3e6e51e6889d4338c37469f9ddb43e5929ef9475
+size 44529096

models/miyazaki_hayao.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8ab0e492efb3b705487db38679e363dc8b1f016692913bbe100587d695a9e2b5
+size 44529096

models/shinkai_makoto.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c3547f611e780e79aebde7f7bc2b6c278555d701620f125583d666351044c486
+size 44529096

requirements_dev.txt

@@ -2,5 +2,4 @@
 black==21.12b0
 flake8==4.0.1
 gradio==2.9.1
-jinja2==3.1.1
-huggingface_hub==0.4.0
+jinja2==3.1.1