Spaces:
Sleeping
Sleeping
File size: 819 Bytes
34bf412 6e342ef 111460e 6e342ef 111460e 6e342ef 111460e 34bf412 6e342ef |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
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)
|