Update app.py

app.py

@@ -86,6 +86,7 @@ torch.backends.cudnn.benchmark = False
 torch.backends.cuda.preferred_blas_library="cublas"
 torch.backends.cuda.preferred_linalg_library="cusolver"
 torch.set_float32_matmul_precision("highest")
+torch.cuda.memory._record_memory_history(enabled='None', context='None', stacks='None', max_entries=9223372036854775807, device=None)
 
 HF_TOKEN = os.getenv("HF_TOKEN")
 FTP_HOST = os.getenv("FTP_HOST")