Update app.py

app.py

@@ -1,5 +1,17 @@
+import pandas as pd
 from transformers import pipeline
 
+# Example data
+data = {
+    'term': [
+        'Atmospheric Chemistry',
+        'Organic Chemistry',
+        'Business Ethics',
+        'Corporate Social Responsibility'
+    ]
+}
+df = pd.DataFrame(data)
+
 # Load the zero-shot classification pipeline
 classifier = pipeline('zero-shot-classification', model='facebook/bart-large-mnli')
 
@@ -15,11 +27,11 @@ def classify_term(term):
 # Classify all terms
 df['classification'] = df['term'].apply(classify_term)
 
-# Recommend discipline for subdisciplines (this part will be more manual unless you have a mapping)
-# Example: Assume subdisciplines are known and mapped to disciplines
+# Example mapping of subdisciplines to disciplines
 subdiscipline_to_discipline = {
     'Atmospheric Chemistry': 'Atmospheric Science',
     'Organic Chemistry': 'Chemistry',
+    'Corporate Social Responsibility': 'Business Ethics'
     # Add your mappings here
 }