TrabajoFinal / app.py
inigo99's picture
Update app.py
e52a1d5
raw
history blame contribute delete
463 Bytes
from huggingface_hub import from_pretrained_fastai
import gradio as gr
from fastai.text.all import *
learner = torch.jit.load("soundWave.pth")
# Definimos una función que se encarga de llevar a cabo las predicciones
def predict(waveform):
pred = learner.predict(waveform)
return {labels[i]}
# Creamos la interfaz y la lanzamos.
gr.Interface(fn=predict, inputs=gr.inputs.Audio(), outputs=gr.outputs.Label(num_top_classes=35)).launch(share=False)