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)