app.py

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",
                           examples = [["def search(arr, low, high, x): if high >= low: mid = (high + low) // 2 if arr[mid] == x: return mid elif arr[mid] > x: return search(arr, low, mid - 1, x) else: return search(arr, mid + 1, high, x) else: return -1"]],
                           allow_flagging="never").launch(inbrowser=True)