Spaces:
Runtime error
Runtime error
import os | |
from models import create_model | |
from PIL import Image | |
from imageio import imwrite | |
import gradio as gr | |
from torchvision import transforms | |
import easydict | |
import torch | |
import numpy as np | |
import cv2 | |
from subprocess import call | |
def run_cmd(command): | |
try: | |
print(command) | |
call(command, shell=True) | |
except KeyboardInterrupt: | |
print("Process interrupted") | |
sys.exit(1) | |
run_cmd("pip install basicsr") | |
run_cmd("pip freeze") | |
from basicsr.archs.rrdbnet_arch import RRDBNet | |
from realesrgan import RealESRGANer | |
from realesrgan.archs.srvgg_arch import SRVGGNetCompact | |
import train | |
esrgan_model = train.build_esrgan() | |
pix2pix_model = train.build_pix2pix(110) | |
def pipeline(img, vivid): | |
img = train.image_preprosses(img, vivid) | |
if type(img) is np.ndarray: | |
img, _ = esrgan_model.enhance(img, outscale=4) | |
img = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2RGB)) | |
img = train.test_pix2pix(img, pix2pix_model) | |
imwrite('filename.png', img) | |
return 'filename.png' | |
description = "μ λλ©μ΄μ ν μ΄λ―Έμ§λ₯Ό λ³Όμ§€μ½ μ€νμΌλ‘ λ°κΎΈλ Pix2pixλͺ¨λΈμ λλ€. μ΄λ―Έμ§ μ λ‘λ ν λμ€λ μ°μΈ‘ μλ¨μ μ°νλͺ¨μ ν΄λ¦ν΄μ μΌκ΅΄ μͺ½μ μ μ¬κ°νμΌλ‘ μλ₯΄κ³ submit νλ©΄ λ©λλ€. ν° μ΄λ―Έμ§λ νμ§ κ°μ μ²΄ν¬ νμκ³ μμ μ΄λ―Έμ§λ μλμΌλ‘ κ°μ λλ μκ΄ X. 볼지μ½μ΄ μΌκ΅΄λ°μ μλ€λ³΄λ μΌκ΅΄μ΄ μ€μμ μκ³ μλ©΄μ΄ ν¬κ² λμ¬μλ‘ μ λλ λλμ λλ€. λ¬Έμ krkmfn@gmail.com " | |
examples=[['momoi_out.png',False]] | |
gr.Interface(pipeline, [gr.inputs.Image(type="pil"),gr.inputs.Checkbox(default=False, label="νμ§ κ°μ ")], gr.outputs.Image(type="file"),description=description,allow_flagging=False,examples=examples,allow_screenshot=False,enable_queue=False).launch() | |