Spaces:
Runtime error
Runtime error
from huggingface_hub import from_pretrained_fastai | |
import gradio as gr | |
from fastai.vision.all import * | |
repo_id = "MarioGL/emotion" | |
learn = load_learner(repo_id) | |
def classify_text(text): | |
pred, pred_idx, probs = learn.predict(text) | |
return pred, probs[pred_idx].item() | |
iface = gr.Interface(fn=classify_text, | |
inputs=gr.Textbox(label="Introduce un texto"), | |
outputs=[gr.Label(), gr.Textbox()], | |
live=True) | |
iface.launch() |