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
|