Update app.py

app.py

@@ -1,4 +1,6 @@
 # import packages
+import shutil
+import os
 __import__('pysqlite3')
 import sys
 sys.modules['sqlite3'] = sys.modules.pop('pysqlite3')
@@ -8,6 +10,14 @@ from datasets import load_dataset
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import gradio as gr
 
+# Function to clear the cache
+def clear_cache(model_name):
+    cache_dir = os.path.expanduser(f'~/.cache/torch/sentence_transformers/{model_name.replace("/", "_")}')
+    if os.path.exists(cache_dir):
+        shutil.rmtree(cache_dir)
+        print(f"Cleared cache directory: {cache_dir}")
+    else:
+        print(f"No cache directory found for: {cache_dir}")
 
 
 # Embedding vector