Spaces:
Runtime error
Runtime error
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) |