Update app.py

app.py

@@ -1,16 +1,11 @@
 import spaces
 import os
 
-os.putenv('PYTORCH_NVML_BASED_CUDA_CHECK','1')
-os.putenv('TORCH_LINALG_PREFER_CUSOLVER','1')
-alloc_conf_parts = [
-    'expandable_segments:True',
-    'pinned_use_background_threads:True'
-]
-
-os.environ['PYTORCH_CUDA_ALLOC_CONF'] = ','.join(alloc_conf_parts)
+os.environ['PYTORCH_NVML_BASED_CUDA_CHECK'] = '1'
+os.environ['TORCH_LINALG_PREFER_CUSOLVER'] = '1'
+os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'expandable_segments:True,pinned_use_background_threads:True'
 os.environ["SAFETENSORS_FAST_GPU"] = "1"
-os.putenv('HF_HUB_ENABLE_HF_TRANSFER','1')
+os.environ['HF_HUB_ENABLE_HF_TRANSFER'] = '1'
 
 import torch