Update app.py

app.py

@@ -10,6 +10,9 @@ print(f' huggingface_hub.__version__ {huggingface_hub.__version__}')
 import functools
 import gc
 import os
+
+os.environ['TORCH_CUDA_ARCH_LIST'] = '9.0'
+
 import subprocess
 
 import shutil