# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("fill-mask", model="casehold/custom-legalbert")

# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("casehold/custom-legalbert")

import gradio as gr
gr.Interface.load("models/casehold/custom-legalbert").launch()