rica's picture
Create app.py
da54da6
raw
history blame
478 Bytes
import gradio as gr
from transformers import pipeline
pipe = pipeline(task="text-classification",
# model that can do 22k-category classification
model="mrm8488/codebert-base-finetuned-detect-insecure-code")
gr.Interface.from_pipeline(pipe,
title="Insecure Code Detector",
description="Label 1 for insecure code",
allow_flagging="never").launch(inbrowser=True)