File size: 667 Bytes
9761238
c910f3a
9b5b908
 
cc86173
c910f3a
9761238
8b76723
38b2509
 
 
 
9761238
9b5b908
c68ca65
b89aa79
9761238
9b5b908
 
 
7b91098
 
a9c809f
9b5b908
9761238
7b91098
7b1f970
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
import gradio as gr
from huggingface_hub import from_pretrained_fastai
from fastai.learner import load_learner
from PIL import Image
from fastai.vision.all import load_learner


repo_id = "JuncalG/Practica1"

learner = from_pretrained_fastai(repo_id)
labels = learner.dls.vocab


def predict(img):
    pred, pred_idx, probs = learner.predict(img)
    return {str(learner.dls.vocab[i]): float(probs[i]) for i in range(len(probs))}

# Usar componentes nuevos (Gradio moderno)
demo = gr.Interface(
    fn=predict,
    inputs=gr.Image(type="pil"), 
    outputs=gr.Label(num_top_classes=5),
    examples=['af3b0115aad1.png', 'b191ba0a2b12.png']
)


demo.launch(share=True)