Update main.py
Browse files
main.py
CHANGED
|
@@ -1,19 +1,3 @@
|
|
| 1 |
-
|
| 2 |
-
# mods = [
|
| 3 |
-
# "diffusers",
|
| 4 |
-
# "torch",
|
| 5 |
-
# "torchvision",
|
| 6 |
-
# "torchaudio",
|
| 7 |
-
# "transformers",
|
| 8 |
-
# "gradio",
|
| 9 |
-
# "soundfile",
|
| 10 |
-
# "progressbar",
|
| 11 |
-
# "librosa",
|
| 12 |
-
# "audiosr",
|
| 13 |
-
# "unidecode",
|
| 14 |
-
# "pyloudnorm",
|
| 15 |
-
# "pydub"]
|
| 16 |
-
|
| 17 |
import gradio as gr
|
| 18 |
import torch
|
| 19 |
import gc # free up memory
|
|
@@ -37,8 +21,6 @@ class AudioUpscaler:
|
|
| 37 |
Upscales audio using the AudioSR model.
|
| 38 |
"""
|
| 39 |
|
| 40 |
-
|
| 41 |
-
@spaces.GPU(duration=300)
|
| 42 |
def __init__(self, model_name="basic", device="auto"):
|
| 43 |
"""
|
| 44 |
Initializes the AudioUpscaler.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
import torch
|
| 3 |
import gc # free up memory
|
|
|
|
| 21 |
Upscales audio using the AudioSR model.
|
| 22 |
"""
|
| 23 |
|
|
|
|
|
|
|
| 24 |
def __init__(self, model_name="basic", device="auto"):
|
| 25 |
"""
|
| 26 |
Initializes the AudioUpscaler.
|