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_0 for secure code, LABEL_1 for insecure code",
                           inputs = gr.inputs.Textbox(placeholder="Type or paste code here", label="Input"),
                           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'],["def foo(request, user): assert user.is_admin, “user does not have access”"]],
                           allow_flagging="never").launch(inbrowser=True)