Spaces:
Sleeping
Sleeping
File size: 547 Bytes
8734633 f709d81 8734633 f709d81 8734633 d337eef |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
import gradio as gr
from EmotionClassifier.EmotionPredictor import EmotionPredictor
from NER_Wrapper.NER_Wrapper import FullNERPipeline
import pandas as pd
emotion_predictor,ner_pipe = EmotionPredictor(),FullNERPipeline()
def predict(description):
ner_text = ner_pipe.process_text(description)
emotions = emotion_predictor(ner_text)
return pd.DataFrame(emotions)
iface = gr.Interface(
fn=predict,
inputs=gr.Textarea(),
outputs="dataframe",
live=False,
title="Emotion Prediction"
)
iface.launch(inline=False) |