update

app.py

@@ -2,11 +2,11 @@ import gradio as gr
 import torchaudio
 from audiocraft.models import MusicGen
 from audiocraft.data.audio import audio_write
-import spaces
+#import spaces
 import logging
 import os 
 import uuid
-from torch.cuda.amp import autocast
+#from torch.cuda.amp import autocast
 import torch
 
 ZERO_GPU_PATCH_TORCH_DEVICE = 1
@@ -18,7 +18,15 @@ logging.info("Carregando o modelo pré-treinado.")
 model = MusicGen.get_pretrained('nateraw/musicgen-songstarter-v0.2')
 model.set_generation_params(duration=30)
 
-@spaces.GPU(duration=120)
+if torch.cuda.is_available():
+    autocast = torch.cuda.amp.autocast
+elif torch.backends.mps.is_available():
+    autocast = torch.amp.autocast
+else:
+    autocast = torch.autocast
+
+
+#@spaces.GPU(duration=120)
 def generate_music(description, melody_audio):
     with autocast():
         logging.info("Iniciando a geração de música.")
@@ -73,4 +81,4 @@ gr.Interface(
         ["trap, synthesizer, songstarters, dark, G# minor, 140 bpm", "./assets/kalhonaho.mp3"],
         ["upbeat, electronic, synth, dance, 120 bpm", None]
     ]
-).launch()
+).launch()

requirements.txt

@@ -1,2 +1,2 @@
 git+https://github.com/facebookresearch/audiocraft#egg=audiocraft
-torchaudio
+#torchaudio