Update main.py

main.py

@@ -1,33 +1,18 @@
 
-import subprocess
-
-mods = [
-"diffusers",
-"torch==2.0.1",
-"torchvision==0.15.2",
-"torchaudio",
-"huggingface_hub",
-"transformers==4.30.2",
-"gradio",
-"soundfile",
-"progressbar",
-"librosa",
-"audiosr",
-"unidecode",
-"pyloudnorm",
-"pydub"]
-
-import importlib
-
-for m in mods:
-    try:
-        importlib.__import__(m.split('=')[0])
-    except Exception as e:
-        print(e)
-        subprocess.run(["uv", "pip", "install", m , "--system"])
-        importlib.__import__(m.split(r'(=|$)')[0])
-        continue
-
+# mods = [
+# "diffusers",
+# "torch",
+# "torchvision",
+# "torchaudio",
+# "transformers",
+# "gradio",
+# "soundfile",
+# "progressbar",
+# "librosa",
+# "audiosr",
+# "unidecode",
+# "pyloudnorm",
+# "pydub"]
 
 import gradio as gr
 import torch
@@ -52,6 +37,8 @@ class AudioUpscaler:
     Upscales audio using the AudioSR model.
     """
 
+
+    @spaces.GPU(duration=300)
     def __init__(self, model_name="basic", device="auto"):
         """
         Initializes the AudioUpscaler.
@@ -394,8 +381,6 @@ class AudioUpscaler:
         # return output_file
 
 
-
-@spaces.GPU(duration=300)
 def inference(audio_file, model_name, guidance_scale, ddim_steps, seed):
     audiosr = build_model(model_name=model_name)