Spaces:
Runtime error
Runtime error
| from upcunet_v3 import RealWaifuUpScaler | |
| import gradio as gr | |
| import time | |
| import logging | |
| import os | |
| from PIL import ImageOps | |
| import numpy as np | |
| import math | |
| def greet(input_img, input_model_name, input_tile_mode): | |
| input_img = np.array(input_img) | |
| if input_model_name not in model_cache: | |
| t1 = time.time() | |
| upscaler = RealWaifuUpScaler(input_model_name[2], ModelPath + input_model_name, half=False, device="cpu") | |
| t2 = time.time() | |
| logger.info(f'load model time, {t2 - t1}') | |
| model_cache[input_model_name] = upscaler | |
| else: | |
| upscaler = model_cache[input_model_name] | |
| logger.info(f'load model from cache') | |
| start = time.time() | |
| result = upscaler(input_img, tile_mode=input_tile_mode) | |
| end = time.time() | |
| logger.info(f'input_model_name, {input_model_name}') | |
| logger.info(f'input_tile_mode, {input_tile_mode}') | |
| logger.info(f'input shape, {input_img.shape}') | |
| logger.info(f'output shape, {result.shape}') | |
| logger.info(f'speed time, {end - start}') | |
| return result | |
| if __name__ == '__main__': | |
| logging.basicConfig(level=logging.INFO, format="[%(asctime)s] [%(process)d] [%(levelname)s] %(message)s") | |
| logger = logging.getLogger() | |
| ModelPath = "weights_v3/" | |
| model_cache = {} | |
| input_model_name = gr.inputs.Dropdown(os.listdir(ModelPath), default="up2x-latest-no-denoise.pth", label='选择model') | |
| input_tile_mode = gr.inputs.Dropdown([0, 1, 2, 3, 4], default=2, label='选择tile_mode') | |
| input_img = gr.inputs.Image(label='image', type='pil') | |
| inputs = [input_img, input_model_name, input_tile_mode] | |
| outputs = "image" | |
| iface = gr.Interface(fn=greet, | |
| inputs=inputs, | |
| outputs=outputs, | |
| allow_screenshot=False, | |
| allow_flagging='never', | |
| examples=[['test-img.jpg', "up2x-latest-no-denoise.pth", 2],['miku1.jpg', "up2x-latest-no-denoise.pth", 2],['ll.jpg', "up2x-latest-no-denoise.pth", 2],['kotori.jpg', "up2x-latest-no-denoise.pth", 2]], | |
| article='感谢大佬开源的项目') | |
| iface.launch() | |