Update app.py

app.py

@@ -1,3 +1,15 @@
-import gradio as gr
 
+
+
+
+# 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()