Update src/ai_classifier.py

src/ai_classifier.py

@@ -1,9 +1,13 @@
-# ai_classifier.py
 from transformers import pipeline
 
-# Load once and reuse
-classifier = pipeline("zero-shot-classification", model="joeddav/xlm-roberta-large-xnli", cache_dir="./hf_cache")
-# Define the candidate labels (in English internally)
+# Load the pipeline once and tell it where to cache the model
+classifier = pipeline(
+    "zero-shot-classification",
+    model="joeddav/xlm-roberta-large-xnli",
+    cache_dir="./hf_cache"  # This line fixes the permission error
+)
+
+# Define the candidate labels
 CATEGORIES = {
     "Family": "కుటుంబం",
     "Friendship": "స్నేహం",
@@ -28,6 +32,7 @@ CATEGORIES = {
 }
 
 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]