Fix classes

app.py

@@ -9,10 +9,11 @@ warnings.filterwarnings("ignore")
 
 learn = load_learner('modelo.pkl')
 
+classes = ['abdomen', 'antebrazo', 'cadera', 'cervical', 'clavicula', 'codo', 'col. torax', 'craneo', 'dedos', 'hombro',
+           'lumbar', 'mano', 'muslo', 'muñeca', 'otros', 'pelvis', 'pierna', 'pies', 'rodilla', 'senos nasales', 'tobillo', 'torax']
+
 # Función de predicción
-def classify_image(image, model=learn):
-    classes = ['abdomen', 'tobillo', 'col. cervical', 'torax', 'clavicula', 'codo', 'pies', 'dedos', 'antebrazo', 'mano', 'cadera',
-               'rodilla', 'pierna', 'col. lumbar', 'otros', 'pelvis', 'hombro', 'senos nasales', 'craneo', 'muslo', 'col. toracica', 'muñeca']
+def classify_image(image, model=learn, classes=classes):
 
     # Cargar imagen y realizar predicción
     img = PILImage.create(image)