Update src/ai_classifier.py

src/ai_classifier.py

@@ -1,13 +1,13 @@
 from transformers import pipeline
 
-# Load the pipeline once and tell it where to cache the model
+# Load the zero-shot-classification pipeline and specify a cache directory.
 classifier = pipeline(
     "zero-shot-classification",
     model="joeddav/xlm-roberta-large-xnli",
-    cache_dir="./hf_cache"  # This line fixes the permission error
+    cache_dir="./hf_cache"
 )
 
-# Define the candidate labels
+# Mapping of English category labels to Telugu.
 CATEGORIES = {
     "Family": "కుటుంబం",
     "Friendship": "స్నేహం",
@@ -35,4 +35,4 @@ def classify_proverb(text):
     """Classifies the proverb and returns the Telugu label."""
     result = classifier(text, list(CATEGORIES.keys()))
     top_label = result["labels"][0]
-    return CATEGORIES[top_label]
+    return CATEGORIES[top_label]