updating

2023-03-16 15:17:57 +01:00
parent f1ad511850
commit ebf9633723
2 changed files with 72 additions and 33 deletions
--- a/src/captcha/ModelDAO.py
+++ b/src/captcha/ModelDAO.py
@@ -4,9 +4,9 @@ import shutil

 class ModelDAO:

-    def saveModel(self, model):
-        shutil.rmtree(model.name, ignore_errors = True)
-        model.save(model.name)
+    def saveModel(self, model, modelFilepath):
+        shutil.rmtree(modelFilepath, ignore_errors = True)
+        model.save(modelFilepath)

    def loadModel(self, modelFilepath):
        return keras.models.load_model(modelFilepath)