Update app.py

app.py

@@ -18,6 +18,7 @@ model.set_generation_params(duration=8)
 
 @spaces.GPU(duration=120)
 def generate_music(description, melody_audio):
+    ZERO_GPU_PATCH_TORCH_DEVICE=1
     logging.info("Iniciando a geração de música.")
     if description:
         description = [description]