Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ os.environ['CUDA_VISIBLE_DEVICES'] = '0'
|
|
16 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
17 |
|
18 |
# Font configuration
|
19 |
-
font_path = '
|
20 |
font_prop = FontProperties(fname=font_path, size=18)
|
21 |
|
22 |
# Model and Processing Parameters
|
@@ -28,7 +28,7 @@ TARGET_SAMPLE_RATE = 16000
|
|
28 |
# Model Initialization
|
29 |
model = TinyVAD(1, 32, 64, patch_size=8, num_blocks=2,
|
30 |
sinc_conv=SINC_CONV, ssm=SSM).to(device)
|
31 |
-
checkpoint_path = '
|
32 |
model.load_state_dict(torch.load(checkpoint_path, weights_only=True))
|
33 |
model.eval()
|
34 |
|
|
|
16 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
17 |
|
18 |
# Font configuration
|
19 |
+
font_path = './Times_New_Roman.ttf'
|
20 |
font_prop = FontProperties(fname=font_path, size=18)
|
21 |
|
22 |
# Model and Processing Parameters
|
|
|
28 |
# Model Initialization
|
29 |
model = TinyVAD(1, 32, 64, patch_size=8, num_blocks=2,
|
30 |
sinc_conv=SINC_CONV, ssm=SSM).to(device)
|
31 |
+
checkpoint_path = './sincvad.ckpt'
|
32 |
model.load_state_dict(torch.load(checkpoint_path, weights_only=True))
|
33 |
model.eval()
|
34 |
|