Spaces:
Sleeping
Sleeping
MUTED64
commited on
Commit
•
47f707a
1
Parent(s):
83a7424
update model
Browse files- 2048_MLP_best-MSE2.7738_ep110.pth +3 -0
- app.py +2 -2
2048_MLP_best-MSE2.7738_ep110.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:92be533b86c7ce80faa2be5e6349e33544dc0a9d40fd66baf97a86664fd1fe36
|
3 |
+
size 3716134
|
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)
|
@@ -61,7 +61,7 @@ with gr.Blocks() as demo:
|
|
61 |
label='Model Path',
|
62 |
value=model_path,
|
63 |
placeholder='Path or URL to the model file',
|
64 |
-
|
65 |
)
|
66 |
with gr.Row():
|
67 |
score = gr.Number(
|
|
|
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)
|
|
|
61 |
label='Model Path',
|
62 |
value=model_path,
|
63 |
placeholder='Path or URL to the model file',
|
64 |
+
interactive=True,
|
65 |
)
|
66 |
with gr.Row():
|
67 |
score = gr.Number(
|