diribes commited on
Commit
2515513
verified
1 Parent(s): a270418

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -14
app.py DELETED
@@ -1,14 +0,0 @@
1
- # repo_id = "YOUR_USERNAME/YOUR_LEARNER_NAME"
2
- repo_id = "diribes/entregable2"
3
-
4
- learner = from_pretrained_fastai(repo_id)
5
- labels = learner.dls.vocab
6
-
7
- # Definimos una funci贸n que se encarga de llevar a cabo las predicciones
8
- def predict(img):
9
- #img = PILImage.create(img)
10
- pred,pred_idx,probs = learner.predict(img)
11
- return {labels[i]: float(probs[i]) for i in range(len(labels))}
12
-
13
- # Creamos la interfaz y la lanzamos.
14
- gr.Interface(fn=predict, inputs=gr.Image(), outputs=gr.Label(num_top_classes=3),examples=['racoon.jpg','weasel.jpg']).launch(share=False)