jethrowang commited on
Commit
634bb0d
·
verified ·
1 Parent(s): 6c23e45

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 = '/share/nas169/jethrowang/fonts/Times_New_Roman.ttf'
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 = '/share/nas169/jethrowang/SincVAD/exp/exp_0.63_tinyvad_psq_0.05/model_epoch_37_val_auroc=0.8894.ckpt'
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