Update app.py

app.py

@@ -1,29 +1,26 @@
-import gradio as gr
 from transformers import pipeline
+import gradio as gr
 
-# Load zero-shot classification model
-classifier = pipeline(
-    "zero-shot-classification",
-    model="MoritzLaurer/deberta-v3-large-zeroshot-v2.0"
-)
+# Load zero-shot classification pipeline
+classifier = pipeline("zero-shot-classification", model="MoritzLaurer/deberta-v3-large-zeroshot-v2.0")
+
+# Define possible classes
+labels = ["dangerous", "mild", "neutral"]
 
-def classify_text(text, labels):
-    labels_list = [label.strip() for label in labels.split(",")]
-    result = classifier(text, candidate_labels=labels_list)
-    formatted = "\n".join([
-        f"{label}: {score:.2f}" for label, score in zip(result["labels"], result["scores"])
-    ])
-    return formatted
+def classify_side_effect(text):
+    if not text.strip():
+        return {"error": "Empty input"}
+    result = classifier(text, candidate_labels=labels)
+    scores = {label: float(score) for label, score in zip(result["labels"], result["scores"])}
+    return scores
 
-demo = gr.Interface(
-    fn=classify_text,
-    inputs=[
-        gr.Textbox(lines=3, label="Text to classify"),
-        gr.Textbox(lines=1, label="Labels (comma separated, e.g. adverse, beneficial, neutral)")
-    ],
-    outputs="text",
-    title="DeBERTa Zero-Shot Classification API",
-    description="Zero-shot classification using MoritzLaurer's DeBERTa-v3-large-zeroshot-v2.0 model."
+# Gradio interface
+iface = gr.Interface(
+    fn=classify_side_effect,
+    inputs=gr.Textbox(label="Enter a side effect or sentence"),
+    outputs=gr.Label(label="Classification Result"),
+    title="Zero-Shot ADR Severity Classifier",
+    description="Classifies a given sentence (e.g. side effect) as dangerous, mild, or neutral using DeBERTa v3 Large Zero-Shot."
 )
 
-demo.launch()
+iface.launch()