fix: Removes previous models

Checks previously downloaded models and removes them before downloading new ones. Fixes the issue #33
Kohulan · Mar 23, 2023 · fb9b858 · fb9b858
1 parent 71acf86
commit fb9b858
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/DECIMER/decimer.py b/DECIMER/decimer.py
@@ -31,7 +31,8 @@
 model_path = str(default_path) + "/DECIMER_model/"
 
 # download models to a default location
-if not os.path.exists(model_path):
+if os.path.exists(model_path):
+    shutil.rmtree(model_path)
     config.download_trained_weights(model_url, default_path)