Spaces:
Build error
Build error
natexcvi
commited on
Commit
•
6ef3be3
1
Parent(s):
3499002
Remove unused code
Browse files- model/model.py +0 -9
model/model.py
CHANGED
@@ -17,15 +17,6 @@ class Model:
|
|
17 |
)
|
18 |
)
|
19 |
|
20 |
-
@classmethod
|
21 |
-
def load_model(cls, model_path: str):
|
22 |
-
custom_objects = {
|
23 |
-
"triplet_loss": cls.triplet_loss_init(0.2),
|
24 |
-
"triplet_accuracy": cls.triplet_accuracy,
|
25 |
-
}
|
26 |
-
with keras.utils.custom_object_scope(custom_objects):
|
27 |
-
return keras.models.load_model(model_path)
|
28 |
-
|
29 |
def predict(self, x: np.ndarray):
|
30 |
return self.trained_model.predict(x)
|
31 |
|
|
|
17 |
)
|
18 |
)
|
19 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
20 |
def predict(self, x: np.ndarray):
|
21 |
return self.trained_model.predict(x)
|
22 |
|