File size: 411 Bytes
da54da6
 
 
67a650b
 
da54da6
 
 
 
1
2
3
4
5
6
7
8
9
import gradio as gr
from transformers import pipeline

pipe = pipeline(task="text-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)