Spaces:
Running
Running
import gradio as gr | |
from transformers import pipeline | |
pipeline = pipeline(task="text-classification", model="Preetham04/sentiment-analysis") | |
def predict(input_text): | |
predictions = pipeline(input_text) | |
print(predictions) | |
result = [] | |
for p in predictions: | |
review = "POSITIVE" if p["label"] == "LABEL_1" else "NEGATIVE" | |
result.append({ | |
"SENTIMENT": review, | |
"SCORE": p["score"] | |
}) | |
return result | |
description = """ | |
The Bot was trained to answer questions based on Clothing. Ask anything! | |
""" | |
gradio_app = gr.Interface( | |
predict, | |
inputs="textbox", | |
outputs="text", | |
title="Sentiment- good or bad?", | |
description=description, | |
examples=[["Loved it!!"], ["Bad quality"]], | |
) | |
if __name__ == "__main__": | |
gradio_app.launch(share=True) | |