MichaelWelsch commited on
Commit
9ef6c6c
·
1 Parent(s): ab9dfa3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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/r10d.pth')
17
 
18
  cluster_som = ClusterSOM()
19
- cluster_som.load("models/cluster_som2.pkl")
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):