TextGuardian / app.py
liyaoshi's picture
Create app.py
7db1e02 verified
raw
history blame
1.1 kB
import requests
import gradio as gr
def moderation(input_text,api_key='sk-proj-KMeL7IIELJPekicYQOYmT3BlbkFJT8CBEbRL5h3OBExmCMHx'):
headers = {
"Content-Type": "application/json",
"Authorization": f"Bearer {api_key}"
}
data = {
"input": input_text
}
response = requests.post('https://api.openai.com/v1/moderations',headers=headers,json=data)
result = response.json()['results'][0]
flag = 1 if result['flagged'] else 0
categories = result['categories']
scores = result['category_scores']
return flag,scores
title = "Smart content moderation"
description = "Check whether your content is potentially harmful."
demo = gr.Interface(
fn=moderation,
inputs=[gr.Text(label='Content',info="Input your text to check")],
outputs=[
gr.Slider(label="Is it harmful?",info="1 is harmful, 0 is safe",minimum=0,maximum=1),
# gr.CheckboxGroup(),
# gr.Text(label="Categories Flag")
gr.Label(label='Categories')
],
title = title,
description = description
)
demo.queue(max_size = 20)
demo.launch(share = True)