| import os | |
| from utils import * | |
| from transformers import pipeline | |
| import gradio as gr | |
| pipe = pipeline("text-classification", model="ShynBui/comment_classification_v2", token=os.environ['HF_TOKEN']) | |
| def classification(text): | |
| classi = pipe(covert_pyVI(text)) | |
| return classi[0]['label'], classi[0]['score'] | |
| iface = gr.Interface(fn=classification, inputs="text", outputs=["text", "text"]) | |
| iface.launch() | |