Spaces:
Runtime error
Runtime error
File size: 709 Bytes
126e43b 8ec1896 9f36cf6 d986501 1a3c90a d986501 1a3c90a 126e43b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
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 requests
API_URL = "https://api-inference.huggingface.co/models/casehold/custom-legalbert"
headers = {"Authorization": "Bearer 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"}
def query(payload):
response = requests.post(API_URL, headers=headers, json=payload)
return response.json()
output = query({
"inputs": "The answer to the universe is [MASK].",
})
import gradio as gr
gr.Interface.load("models/casehold/custom-legalbert").launch() |