Gradio2

app.py

@@ -8,7 +8,8 @@ import cv2
 
 model = from_pretrained_keras('Rietta/CycleGAN_DL', compile=False)
 
-imagen = cv2.imread('Kirby.png', cv2.COLOR_BGR2RGB) 
+imagen = cv2.imread('Kirby.png') 
+RGB_img = cv2.cvtColor(imagen, cv2.COLOR_BGR2RGB)
 
 def transform(img, direction):
     img = (img / 127.5) - 1
@@ -17,7 +18,7 @@ def transform(img, direction):
     elif direction == 1:
         pred = model.generator_sims.predict(img[None,:,:,:])[0]
     else:
-        pred = imagen
+        pred = RGB_img
     pred = (pred-pred.min())/(pred.max()-pred.min())
     pred = (pred * 255).astype(np.uint8)
     return pred