pushing UI updates

app.py

@@ -1,9 +1,31 @@
 import gradio as gr
+from transformers import pipeline
 
+classifier_pipe = pipeline("text-classification", model="LennardZuendorf/legalis-BERT", top_k=None)
 
-def greet(name):
-    return "Hello " + name + "!!"
 
+def predict_fun(text):
+    predictions = classifier_pipe(text)
+    return {p["label"]: p["score"] for p in predictions[0]}
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+
+with gr.Blocks(title='Legalis') as interface:
+    with gr.Row():
+        gr.Markdown(
+            """
+            # Legalis BERT Demo
+            Start typing below to see the output.
+            """)
+    with gr.Row():
+        with gr.Column():
+            input_text = gr.Textbox(label="Case Facts")
+            with gr.Row():
+                predict = gr.Button("Predict")
+        with gr.Column():
+            label = gr.Label(label="Predicted Winner")
+    with gr.Row():
+        interpretation = gr.components.Interpretation(input_text, visible=False)
+
+    predict.click(predict_fun, input_text, label)
+
+interface.launch()

requirements.txt

@@ -1 +1,2 @@
-gradio
+gradio
+transformers