Spaces:
Runtime error
Runtime error
| 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) |