mohd43's picture
Update app.py
482f94f verified
raw
history blame
444 Bytes
from transformers import pipeline
pipeline = pipeline("text-classification" , model = "ProsusAI/finbert")
def predict(input_text):
predictions = pipeline(input_text )
print(predictions)
return predictions[0]
gradio_app = gr.Interface(
predict,
inputs=gr.Textbox(label="Write a text"),
outputs=gr.Textbox(label="output a text"),
title="Financial Sentiment Analysis",
live=True,
allow_flagging="never",
)
gradio_app.launch()