File size: 576 Bytes
4530bc5
 
 
6dbb309
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16

from fastai.vision.all import *

# Esta es la clave: el modelo necesita que estas funciones EXISTAN
# pero no pueden depender de ds_combined porque no lo tenemos en el servidor.

def get_x(i):
    # Si 'i' ya es una imagen (lo que pasa en la App), la devolvemos
    if isinstance(i, (PILImage, Image.Image)): return i
    # Si el modelo intenta buscar un índice durante la carga, devolvemos algo vacío
    return PILImage.create(np.zeros((126,126,3), dtype=np.uint8))

def get_y(i):
    # El modelo solo necesita saber que esto devuelve una categoría
    return "Unknown"