better check

app.py

@@ -53,6 +53,7 @@ import spaces # for Hugging Face ZeroGPU deployment
 import re
 import plotly.graph_objects as go
 from numpy.lib.user_array import container
+import shutil
 
 # Only for Hugging Face hosting - Add the Hugging Face cache to persistent storage to avoid downloading safetensors every time the demo sleeps and wakes up
 os.environ['HF_HOME'] = '/data/.huggingface'
@@ -113,7 +114,8 @@ class ShapeWordsDemo:
         print(f"Using device: {device} in initialize_models")
 
         # Download Shape2CLIP code if it doesn't exist
-        if not os.path.exists("shapewords_paper_code"):
+        if not os.path.exists("shapewords_paper_code/geometry_guidance_models.py"):
+            shutil.rmtree("shapewords_paper_code/", ignore_errors=True)
             print("Loading models file")
             os.system("git clone https://github.com/lodurality/shapewords_paper_code.git")