SFW-Converter / app.py
amirgame197's picture
Create app.py
8637ba8 verified
raw
history blame
300 Bytes
import gradio as gr
from nudenet import NudeDetector
def process(input_img):
detector = NudeDetector(model_path="640m.onnx", inference_resolution=640)
converted = detector.censor(input_img)
return converted
iface = gr.Interface(fn=process,inputs="image", outputs="image")
iface.launch()