Update app.py

app.py

@@ -82,6 +82,7 @@ def generate(
             
         else:
             pipe.to(device)
+            pipe.enable_model_cpu_offload()
         
     generator = torch.Generator().manual_seed(seed)