Ronicola commited on
Commit
8841c41
1 Parent(s): 775bfbb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,9 +8,9 @@ learner = from_pretrained_fastai(repo_id)
8
  labels = learner.dls.vocab
9
 
10
  # Definimos una función que se encarga de llevar a cabo las predicciones
11
- def predict(img):
12
  #img = PILImage.create(img)
13
- pred,pred_idx,probs = learner.predict(img)
14
  return {labels[i]: float(probs[i]) for i in range(len(labels))}
15
 
16
  # Creamos la interfaz y la lanzamos.
 
8
  labels = learner.dls.vocab
9
 
10
  # Definimos una función que se encarga de llevar a cabo las predicciones
11
+ def predict(predictThis):
12
  #img = PILImage.create(img)
13
+ pred,pred_idx,probs = learner.predict(predictThis)
14
  return {labels[i]: float(probs[i]) for i in range(len(labels))}
15
 
16
  # Creamos la interfaz y la lanzamos.