Update server.py

server.py

@@ -9,6 +9,18 @@ from ROBERTAmodel import *
 from BERTmodel import *
 from DISTILLBERTmodel import *
 
+import shutil
+import os
+
+CACHE_DIR = "./hf_cache"
+if os.path.exists(CACHE_DIR):
+    try:
+        shutil.rmtree(CACHE_DIR)
+        print("✅ Cleared hf_cache directory")
+    except Exception as e:
+        print("❌ Failed to clear hf_cache:", e)
+
+        
 VISUALIZER_CLASSES = {
     "BERT": BERTVisualizer,
     "RoBERTa": RoBERTaVisualizer,