vruizext commited on
Commit
e1f891e
1 Parent(s): 22bc3e8

Fix missing var

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ warnings.filterwarnings("ignore")
10
  learn = load_learner('modelo.pkl')
11
 
12
  # Funci贸n de predicci贸n
13
- def classify_image(image):
14
  classes = ['abdomen', 'tobillo', 'col. cervical', 'torax', 'clavicula', 'codo', 'pies', 'dedos', 'antebrazo', 'mano', 'cadera',
15
  'rodilla', 'pierna', 'col. lumbar', 'otros', 'pelvis', 'hombro', 'senos nasales', 'craneo', 'muslo', 'col. toracica', 'mu帽eca']
16
 
 
10
  learn = load_learner('modelo.pkl')
11
 
12
  # Funci贸n de predicci贸n
13
+ def classify_image(image, model=learn):
14
  classes = ['abdomen', 'tobillo', 'col. cervical', 'torax', 'clavicula', 'codo', 'pies', 'dedos', 'antebrazo', 'mano', 'cadera',
15
  'rodilla', 'pierna', 'col. lumbar', 'otros', 'pelvis', 'hombro', 'senos nasales', 'craneo', 'muslo', 'col. toracica', 'mu帽eca']
16