Spaces:
Build error
Build error
Commit
·
9ef6c6c
1
Parent(s):
ab9dfa3
Update app.py
Browse files
app.py
CHANGED
@@ -13,10 +13,10 @@ from funcs.dataloader import BaseDataset2, read_json_files
|
|
13 |
|
14 |
DEVICE = torch.device("cpu")
|
15 |
reducer10d = PHATEAE(epochs=30, n_components=10, lr=.0001, batch_size=128, t='auto', knn=8, relax=True, metric='euclidean')
|
16 |
-
reducer10d.load('models/
|
17 |
|
18 |
cluster_som = ClusterSOM()
|
19 |
-
cluster_som.load("models/
|
20 |
|
21 |
# ml inference
|
22 |
def get_som_mp4(file, slice_select, reducer=reducer10d, cluster=cluster_som):
|
|
|
13 |
|
14 |
DEVICE = torch.device("cpu")
|
15 |
reducer10d = PHATEAE(epochs=30, n_components=10, lr=.0001, batch_size=128, t='auto', knn=8, relax=True, metric='euclidean')
|
16 |
+
reducer10d.load('models/r10d_3.pth')
|
17 |
|
18 |
cluster_som = ClusterSOM()
|
19 |
+
cluster_som.load("models/cluster_som3.pkl")
|
20 |
|
21 |
# ml inference
|
22 |
def get_som_mp4(file, slice_select, reducer=reducer10d, cluster=cluster_som):
|