Gradio2

app.py

@@ -8,7 +8,7 @@ import cv2
 
 model = from_pretrained_keras('Rietta/CycleGAN_DL', compile=False)
 
-imagen = cv2.imread('Kirby.png', cv2.COLOR_BGR2RGB) 
+imagen = cv2.imread('Kirby.png', cv2.COLOR_RGB2BGR) 
 
 def transform(img, direction):
     img = (img / 127.5) - 1
@@ -22,16 +22,11 @@ def transform(img, direction):
     pred = (pred * 255).astype(np.uint8)
     return pred
 
-#examples_gta = [[path, 'GTA->REAL'] for path in glob('Examples/gta*')]
-#examples_real = [[path, 'REAL->GTA'] for path in glob('Examples/real*')]
-#examples = [*examples_gta, *examples_real]
-
 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'))
-                    #examples=examples)
 
 if __name__ == '__main__':
     demo.launch()