Spaces:
Sleeping
Sleeping
MUTED64
commited on
Commit
•
7ca56ed
1
Parent(s):
47f707a
update model
Browse files- 3072_MLP_best-MSE1.6230_ep23.pth +3 -0
- 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 = "./
|
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)
|