Update app.py

app.py

@@ -4,6 +4,9 @@ from audiosr import super_resolution, build_model
 import torch
 import gc # free up memory
 
+torch.cuda.set_per_process_memory_fraction(0.95, 0)  # This reserves 95% of GPU memory for this process
+torch.backends.cuda.matmul.allow_tf32 = True  # Optional: Enable TensorFloat-32 mode for efficient memory usage
+torch.cuda.set_max_split_size_mb(64)  # Adjust 64 MB to suit your needs based on experimentation
 
 # @spaces.GPU(duration=180)
 def inference(audio_file, model_name, guidance_scale, ddim_steps, seed):