Spaces:
Sleeping
Sleeping
roychao19477
commited on
Commit
·
0f866bf
1
Parent(s):
623bfa0
Update module
Browse files
app.py
CHANGED
|
@@ -71,8 +71,10 @@ from config import sampling_rate
|
|
| 71 |
import spaces
|
| 72 |
|
| 73 |
# Load model once globally
|
| 74 |
-
ckpt_path = "ckpts/ep215_0906.oat.ckpt"
|
| 75 |
-
model = AVSEModule.load_from_checkpoint(ckpt_path)
|
|
|
|
|
|
|
| 76 |
model.to("cuda")
|
| 77 |
model.eval()
|
| 78 |
|
|
|
|
| 71 |
import spaces
|
| 72 |
|
| 73 |
# Load model once globally
|
| 74 |
+
#ckpt_path = "ckpts/ep215_0906.oat.ckpt"
|
| 75 |
+
#model = AVSEModule.load_from_checkpoint(ckpt_path)
|
| 76 |
+
state_dict = torch.load("ckpts/ep215_0906.oat.ckpt")
|
| 77 |
+
model.load_state_dict(state_dict, strict=True)
|
| 78 |
model.to("cuda")
|
| 79 |
model.eval()
|
| 80 |
|