File size: 476 Bytes
cdad8df
7f61b34
cdad8df
93f6be7
 
cdad8df
 
 
 
 
93f6be7
 
482f94f
 
cdad8df
 
482f94f
 
93f6be7
 
482f94f
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
from transformers import pipeline
import gradio as gr
pipe = pipeline("text-classification" , model = "ProsusAI/finbert")

def predict(input_text):
  predictions = pipe(input_text,top_k=3 )
  out = {}
  for item in predictions: 
    out[item["label"]] = item["score"]
  return out

gradio_app = gr.Interface(
  predict,
  inputs=gr.Textbox(label="Write a text"),
  outputs=gr.Label(),
  title="Sentiment Analysis",
  live=True,
  allow_flagging="never",
)

gradio_app.launch()