Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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(
|
12 |
#img = PILImage.create(img)
|
13 |
-
pred,pred_idx,probs = learner.predict(
|
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.
|