Update app.py

app.py

@@ -3,6 +3,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
 import numpy as np
 
+# Load model from the Hub
 model = AutoModelForSequenceClassification.from_pretrained("your-username/roberta-adr-classifier")
 tokenizer = AutoTokenizer.from_pretrained("your-username/roberta-adr-classifier")
 
@@ -17,9 +18,12 @@ def classify_adr(text):
         "Severe (1)": float(probs[1])
     }
 
-demo = gr.Interface(fn=classify_adr, 
-                    inputs=gr.Textbox(lines=4, label="ADR Description"), 
-                    outputs=gr.Label(num_top_classes=2),
-                    title="ADR Severity Classifier")
+demo = gr.Interface(
+    fn=classify_adr,
+    inputs=gr.Textbox(lines=4, label="ADR Description"),
+    outputs=gr.Label(num_top_classes=2),
+    title="ADR Severity Classifier",
+    description="Paste a short description of a suspected adverse drug reaction and the model will assess its severity."
+)
 
 demo.launch()