unilight commited on
Commit
b7ad4c6
1 Parent(s): 7669465

fix sslmos

Browse files
Files changed (2) hide show
  1. app.py +0 -2
  2. models/sslmos.py +2 -0
app.py CHANGED
@@ -1,5 +1,4 @@
1
  import os
2
- from glob import glob
3
 
4
  import torch.nn.functional as F
5
  import torchaudio
@@ -7,7 +6,6 @@ from loguru import logger
7
  import gradio as gr
8
 
9
  from huggingface_hub import hf_hub_download
10
- import time
11
  import torch
12
  import yaml
13
 
 
1
  import os
 
2
 
3
  import torch.nn.functional as F
4
  import torchaudio
 
6
  import gradio as gr
7
 
8
  from huggingface_hub import hf_hub_download
 
9
  import torch
10
  import yaml
11
 
models/sslmos.py CHANGED
@@ -35,6 +35,8 @@ class SSLMOS(torch.nn.Module):
35
  decoder_dnn_dim: int = 64,
36
  output_type: str = "scalar",
37
  range_clipping: bool = True,
 
 
38
  ):
39
  super().__init__() # this is needed! or else there will be an error.
40
  self.use_mean_listener = use_mean_listener
 
35
  decoder_dnn_dim: int = 64,
36
  output_type: str = "scalar",
37
  range_clipping: bool = True,
38
+ # dummy
39
+ num_domains: int = None,
40
  ):
41
  super().__init__() # this is needed! or else there will be an error.
42
  self.use_mean_listener = use_mean_listener