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\n\nThis app identifies whether the input code is vulnerable to cyber attacks that may cause harm to systems such as theft or loss of valuable data. (Other examples of code in examples.txt under Files and versions)",
                           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)