Update app.py

app.py

@@ -7,6 +7,7 @@ import copy
 from pathlib import Path
 import secrets
 import torch
+import gc
 import os
 import io
 from io import BytesIO