Update app.py

app.py

@@ -3,7 +3,6 @@ import gradio as gr
 from skimage.io import imsave
 from skimage.color import lab2rgb
 from tensorflow.keras.utils import img_to_array, load_img
-from PIL import Image
 
 repo_id="saruizn/trabajo"
 
@@ -11,7 +10,7 @@ model = from_pretrained_keras(repo_id)
 
 def predict(img):
     partes=[]
-    imsave("input.jpg",Image.fromarray((img * 255).astype(np.uint8)))
+    imsave("input.jpg",(img * 255).astype(np.uint8))
     imag = img_to_array(load_img("input.jpg"))
     partes=divideImage(imag)
     i=0