norbench_SA / app.py
AnnaPalatkina's picture
m
9761446
raw
history blame
1.04 kB
from sentiment_wrapper import PredictionModel
import gradio as gr
model = PredictionModel()
def predict(text:str):
result = model.predict([text])[0]
return f'class: {result}'
markdown_text = '''
<br>
<br>
This space provides a gradio demo and an easy-to-run wrapper of the pre-trained model for fine-grained sentiment analysis in Norwegian language, pre-trained on the [NoReC dataset](https://huggingface.co/datasets/norec).
The model can be easily used for predicting sentiment as follows:
```python
>>> from sentiment_wrapper import PredictionModel
>>> model = PredictionModel()
>>> model.predict(['vi liker svart kaffe'])
[2]
```
'''
with gr.Blocks() as demo:
with gr.Row(equal_height=False) as row:
text_input = gr.Textbox(label="input")
text_output = gr.Textbox(label="output")
with gr.Row(scale=4) as row:
text_button = gr.Button("submit").style(full_width=True)
text_button.click(fn=predict, inputs=text_input, outputs=text_output)
gr.Markdown(markdown_text)
demo.launch()