openai-detector / app.py
WUJUNCHAO's picture
Update app.py
8ec8428
import gradio as gr
from transformers import pipeline
pipe = pipeline("text-classification", model="roberta-base-openai-detector")
def greet(text):
data = pipe(text)[0]
label = data["label"]
score = data["score"]
score = score * 100
return f"The text is {label}, with {score:.1f}% possibility."
iface = gr.Interface(fn=greet, inputs="text", outputs="text")
iface.launch()