Update app.py

app.py

@@ -1,4 +1,3 @@
-
 try:
     import spaces  # Para ambientes específicos como Hugging Face Spaces
     USING_SPACES = True
@@ -11,48 +10,33 @@ import torchaudio
 from functools import partial
 from resemble_enhance.enhancer.inference import denoise, enhance
 
-
-
-
 def gpu_decorator(func):
-    def wrapper(*args, **kwargs):
-        device = "cpu"
-        if not USING_SPACES and torch.cuda.is_available():
-            device = "cuda"
-        kwargs['device'] = device
-        return func(*args, **kwargs)
-    return wrapper
-
+    if USING_SPACES:
+        return spaces.GPU(func)
+    else:
+        return func
 
 @gpu_decorator
-def _fn(path, solver, nfe, tau, denoising, unlimited, device="cpu"):
+def _fn(path, solver, nfe, tau, denoising, unlimited):
     if path is None:
         return "Please upload an audio file.", None
-
     info = torchaudio.info(path)
     if not unlimited and (info.num_frames / info.sample_rate > 60):
         return "Only audio files shorter than 60 seconds are supported.", None
-
     solver = solver.lower()
     nfe = int(nfe)
     lambd = 0.9 if denoising else 0.1
-
     dwav, sr = torchaudio.load(path)
     dwav = dwav.mean(dim=0)
-
-    wav1, new_sr = denoise(dwav, sr, device)
-    wav2, new_sr = enhance(dwav, sr, device, nfe=nfe, solver=solver, lambd=lambd, tau=tau)
-
+    wav1, new_sr = denoise(dwav, sr)  # Remove o argumento device
+    wav2, new_sr = enhance(dwav, sr, nfe=nfe, solver=solver, lambd=lambd, tau=tau)  # Remove o argumento device
     wav1 = wav1.cpu().numpy()
     wav2 = wav2.cpu().numpy()
-
     return (new_sr, wav1), (new_sr, wav2)
 
 
-def create_app():
-    """
-    Cria a interface do Gradio.
-    """
+
+with gr.Blocks() as app:
     inputs = [
         gr.Audio(type="filepath", label="Input Audio"),
         gr.Dropdown(
@@ -83,12 +67,10 @@ def create_app():
             label="Allow Unlimited Audio Length (supports files longer than 60 seconds)",
         ),
     ]
-
     outputs = [
         gr.Audio(label="Output Denoised Audio"),
         gr.Audio(label="Output Enhanced Audio"),
     ]
-
     # Interface Gradio
     interface = gr.Interface(
         fn=_fn,
@@ -97,23 +79,17 @@ def create_app():
         inputs=inputs,
         outputs=outputs,
     )
-
-    return interface.queue()  # Ativa filas para processamento
+    app = interface.queue()
 
 
-def main(port=7860, host="0.0.0.0", share=False, api=False):
-    """
-    Inicializa a aplicação dependendo do ambiente.
-    """
+def main():
     global app
     print("Starting app...")
-    app = create_app()
-    app.launch(server_name=host, server_port=port, share=share, show_api=api)
+    app.launch(share=True)
 
 
 if __name__ == "__main__":
     if not USING_SPACES:
         main()
     else:
-        app = create_app()
-        app.launch()
+        app.launch(share=True)