1

app.py

@@ -0,0 +1,34 @@
+import gradio as gr
+from transformers import pipeline
+
+classifier = pipeline(
+    "text-generation",
+    model="AtrriJi/smolified-risk-clause-classifier",
+    device=-1  # force CPU
+)
+
+def classify(clause):
+    prompt = f"""
+Classify the legal clause into a category and risk level.
+Return JSON with keys category and risk_level.
+
+Clause:
+{clause}
+"""
+
+    output = classifier(
+        prompt,
+        max_new_tokens=100,
+        do_sample=False
+    )[0]["generated_text"]
+
+    return output
+
+demo = gr.Interface(
+    fn=classify,
+    inputs="text",
+    outputs="text",
+    title="Risk Clause Classifier"
+)
+
+demo.launch()