MUTED64 commited on
Commit
7ca56ed
1 Parent(s): 47f707a

update model

Browse files
Files changed (2) hide show
  1. 3072_MLP_best-MSE1.6230_ep23.pth +3 -0
  2. app.py +1 -1
3072_MLP_best-MSE1.6230_ep23.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ba9045de8bf52503dbb799446da864d2046882f70f546e4bc6342ea3f9d9af6d
3
+ size 3716120
app.py CHANGED
@@ -6,7 +6,7 @@ from waifu_scorer.mlp import MLP
6
  import clip
7
 
8
  # Load the pre-trained model
9
- model_path = "./2048_MLP_best-MSE2.7738_ep110.pth"
10
  device = "cpu"
11
  dtype = torch.float32
12
  s = torch.load(model_path, map_location=device)
 
6
  import clip
7
 
8
  # Load the pre-trained model
9
+ model_path = "./3072_MLP_best-MSE1.6230_ep23.pth"
10
  device = "cpu"
11
  dtype = torch.float32
12
  s = torch.load(model_path, map_location=device)