Spaces:
Runtime error
Runtime error
from random import choices | |
import numpy as np | |
import gradio as gr | |
from glob import glob | |
from huggingface_hub import from_pretrained_keras | |
import cv2 | |
model = from_pretrained_keras('Rietta/CycleGAN_DL', compile=False) | |
imagen = cv2.imread('Kirby.png') | |
RGB_img = cv2.cvtColor(imagen, cv2.COLOR_BGR2RGB) | |
def transform(img, direction): | |
img = (img / 127.5) - 1 | |
if direction==0: | |
pred = model.generator_wow.predict(img[None,:,:,:])[0] | |
elif direction == 1: | |
pred = model.generator_sims.predict(img[None,:,:,:])[0] | |
else: | |
pred = RGB_img | |
pred = (pred-pred.min())/(pred.max()-pred.min()) | |
pred = (pred * 255).astype(np.uint8) | |
return pred | |
demo = gr.Interface(fn=transform, | |
inputs=[gr.inputs.Image(shape=(256, 256), type='numpy'), | |
gr.inputs.Radio(choices=['Sims', 'Warcraft', "Kirby"], | |
type='index')], | |
outputs=gr.outputs.Image(type='numpy')) | |
if __name__ == '__main__': | |
demo.launch() |