Hugging Face's logo

Spaces:
tori29umai
/
Photo2ShojoManga
Running on Zero

App Files Community
Photo2ShojoManga
File size: 2,903 Bytes 
3542be4
 
 
 
 
 
 
 
 
 
31877a7
5836c22
 
e663a52
5836c22
 
 
 
 
 
 
 
 
 
 
3542be4
31877a7
3542be4
 
 
 
 
 
 
31877a7
111970c
63144e1
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
3542be4
 
31877a7
9c9bde8
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
5836c22
9c9bde8
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
 
import os

import requests
from tqdm import tqdm
import shutil

from PIL import Image, ImageOps
import numpy as np
import cv2

def dl_cn_model(model_dir):
    folder = model_dir
    file_name = 'diffusion_pytorch_model.safetensors'
    url = "https://huggingface.co/2vXpSwA7/iroiro-lora/resolve/main/test_controlnet2/CN-anytest_v4-marged.safetensors"
    file_path = os.path.join(folder, file_name)
    if not os.path.exists(file_path):
        response = requests.get(url, allow_redirects=True)
        if response.status_code == 200:
            with open(file_path, 'wb') as f:
                f.write(response.content)
            print(f'Downloaded {file_name}')
        else:
            print(f'Failed to download {file_name}')
    else:
        print(f'{file_name} already exists.')

def dl_cn_config(model_dir):
  folder = model_dir
  file_name = 'config.json'
  file_path = os.path.join(folder, file_name)
  if not os.path.exists(file_path):
     config_path = os.path.join(os.getcwd(), file_name)
     shutil.copy(config_path, file_path)

def dl_tagger_model(model_dir):
    model_id = 'SmilingWolf/wd-vit-tagger-v3'
    files = [
        'config.json', 'model.onnx', 'selected_tags.csv', 'sw_jax_cv_config.json'
    ]

    if not os.path.exists(model_dir):
        os.makedirs(model_dir)
    
    for file in files:
        file_path = os.path.join(model_dir, file)
        if not os.path.exists(file_path):
            url = f'https://huggingface.co/{model_id}/resolve/main/{file}'
            response = requests.get(url, allow_redirects=True)
            if response.status_code == 200:
                with open(file_path, 'wb') as f:
                    f.write(response.content)
                print(f'Downloaded {file}')
            else:
                print(f'Failed to download {file}')
        else:
            print(f'{file} already exists.')


def dl_lora_model(model_dir):
    models = {
        "syoujomannga_line.safetensors": "https://huggingface.co/tori29umai/Egara_Lora/resolve/main/syoujomannga_line.safetensors",
        "tori29umai_line.safetensors": "https://huggingface.co/tori29umai/Egara_Lora/resolve/main/tori29umai_line.safetensors",
    # "gekiga_line.safetensors": "https://huggingface.co/tori29umai/Egara_Lora/resolve/main/gekiga_line.safetensors"
    }
    
    # 各モデルを確認し、必要に応じてダウンロード
    for file_name, url in models.items():
        file_path = os.path.join(model_dir, file_name)
        if not os.path.exists(file_path):
            response = requests.get(url, allow_redirects=True)
            if response.status_code == 200:
                with open(file_path, 'wb') as f:
                    f.write(response.content)
                print(f'Downloaded {file_name}')
            else:
                print(f'Failed to download {file_name}')
        else:
            print(f'{file_name} already exists.')