Update app.py

app.py

@@ -17,6 +17,7 @@ from functools import partial
 import io
 from io import BytesIO
 
+torch.cuda.empty_cache()
 
 subprocess.run(shlex.split('pip install wheel/torchmcubes-0.1.0-cp310-cp310-linux_x86_64.whl'))