Update server.py

server.py

@@ -7,6 +7,15 @@ import os
 from PIL import Image
 from transformers import CLIPProcessor, CLIPModel
 import numpy as np
+import shutil
+import os
+from transformers import CLIPProcessor, CLIPModel
+
+# Clear the cache before loading the model
+cache_dir = os.path.expanduser("~/.cache/huggingface")
+if os.path.exists(cache_dir):
+    shutil.rmtree(cache_dir)
+    print("Cache cleared.")
 
 # Define the MIDM model
 class MIDM(nn.Module):