Spaces:
Sleeping
Sleeping
unilight
commited on
Commit
·
b7ad4c6
1
Parent(s):
7669465
fix sslmos
Browse files- app.py +0 -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
|