Update app.py

app.py

@@ -2,9 +2,12 @@ import os
 import gradio as gr
 import glob
 
-os.system("pip install gsutil")
+os.system("!apt-get update -qq && apt-get install -qq libfluidsynth1 build-essential libasound2-dev libjack-dev")
 
+# install t5x
 os.system("git clone --branch=main https://github.com/google-research/t5x")
+# pin T5X for python 3.7 compatibility
+os.system("cd t5x; git reset --hard 2e05ad41778c25521738418de805757bf2e41e9e; cd ..")
 os.system("mv t5x t5x_tmp; mv t5x_tmp/* .; rm -r t5x_tmp")
 os.system("sed -i 's:jax\[tpu\]:jax:' setup.py")
 os.system("python3 -m pip install -e .")
@@ -14,10 +17,9 @@ os.system("git clone --branch=main https://github.com/magenta/mt3")
 os.system("mv mt3 mt3_tmp; mv mt3_tmp/* .; rm -r mt3_tmp")
 os.system("python3 -m pip install -e .")
 
-# copy checkpoints
-os.system("gsutil -q -m cp -r gs://mt3/checkpoints .")
+os.system("pip install jax[cuda]==0.3.15 -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html")
 
-# copy soundfont (originally from https://sites.google.com/site/soundfonts4u)
+os.system("gsutil -q -m cp -r gs://mt3/checkpoints .")
 os.system("gsutil -q -m cp gs://magentadata/soundfonts/SGM-v2.01-Sal-Guit-Bass-V1.3.sf2 .")
 
 import functools