import gradio as gr from transformers import pipeline model = 'mrm8488/electricidad-small-finetuned-amazon-review-classification' def classify(text): cls= pipeline("text-classification", model=model) return cls(text)[0]['label'] + '➗ ⭐⭐⭐⭐⭐' gr.Interface(fn=classify, inputs=["textbox"], outputs="text").launch(debug=True)