File size: 1,089 Bytes
70056da 3f46b35 70056da ce4e568 70056da bcd931a 70056da bcd931a 70056da bcd931a 70056da bcd931a 70056da 8ea15a5 777e2a0 70056da |
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 |
import pandas as pd
from fastai.text.all import *
from datasets import load_dataset
from transformers import pipeline
import gradio as gr
# Modelo de clasificaci贸n usando LSTM
#learner = load_learner('modelLSTM.pkl')
# Modelo de clasificaci贸n de texto usando modelos de lenguaje
#learner = load_learner('modelML.pkl')
# Modelo de clasificaci贸n basados en mecanismos de atenci贸n
classifier = pipeline('text-classification', model='edgilr/clasificador-rotten-tomatoes')
def predict(txt):
# Modelo de clasificaci贸n usando LSTM o modelo de clasificaci贸n de texto usando modelos de lenguaje
#pred,pred_idx,probs = learner.predict(txt)
#return pred
# Modelo de clasificaci贸n basados en mecanismos de atenci贸n
return classifier(txt)[0]['label']
gr.Interface(fn=predict, inputs="text", outputs="text",
examples=['the story gives ample opportunity for large-scale action and suspense , which director shekhar kapur supplies with tremendous skill .',
'the thing looks like a made-for-home-video quickie .']).launch(share=True) |