MUTED64 commited on
Commit
47f707a
1 Parent(s): 83a7424

update model

Browse files
Files changed (2) hide show
  1. 2048_MLP_best-MSE2.7738_ep110.pth +3 -0
  2. 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 = "./1024_MLP_best-MSE4.1636_ep75.pth"
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
- # interactive=not fix_model_path,
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(