Edupy's picture
Arquitectura modular con funciones get_x/y reales
6dbb309
raw
history blame contribute delete
576 Bytes
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"