Update model_utils.py

model_utils.py

@@ -1,18 +1,23 @@
 import os
-os.environ["TRANSFORMERS_CACHE"] = "./cache"
-os.environ["HF_HOME"] = "./cache"  # also set HF_HOME to cache in same dir
+
+# Fix: Redirect Hugging Face cache to a writable folder
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
+os.environ["HF_HOME"] = "/tmp/hf_cache"  # new standard from transformers v5+
 
 from transformers import pipeline
 
+# Initialize zero-shot classification pipeline
 classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
 
+# Define labels for classification
 labels = ["question", "option", "answer", "other"]
 
 def classify_chunk(text):
     result = classifier(text, labels)
-    return result['labels'][0]  # Most likely label
+    return result['labels'][0]  # Return the top predicted label
 
 def extract_mcqs_with_model(text):
+    # Split text into chunks, skipping empty ones
     chunks = [chunk.strip() for chunk in text.split("\n\n") if chunk.strip()]
     mcqs = []
     current = {"question": "", "options": [], "answer": ""}