Spaces:
Runtime error
Runtime error
File size: 310 Bytes
47c1e73 |
1 2 3 4 5 6 7 8 9 10 |
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() |