Spaces:
Runtime error
Runtime error
import gradio as gr | |
from transformers import pipeline | |
pipeline = pipeline(task="text-classification", model="Preetham04/sentiment-analysis_tf") | |
def predict(input_text): | |
predictions = pipeline(input_text) | |
result = [] | |
for p in predictions: | |
review = "POSITIVE" if p["label"] == "LABEL_1" else "NEGATIVE" | |
result.append({ | |
"review": review, | |
"score": p["score"] | |
}) | |
return result | |
gradio_app = gr.Interface( | |
predict, | |
inputs="textbox", | |
outputs="text", | |
title="Sentiment- good or bad?", | |
) | |
if __name__ == "__main__": | |
gradio_app.launch(share=True) |