Update app.py

app.py

@@ -248,6 +248,9 @@ def call_pipe(
     # 7. Denoising loop
     num_warmup_steps = len(timesteps) - num_inference_steps * pipe.scheduler.order
     pipe._num_timesteps = len(timesteps)
+    pipe.text_encoder.to("cpu") 
+    pipe.text_encoder_2.to("cpu")  
+    torch.cuda.empty_cache() 
     
     with pipe.progress_bar(total=num_inference_steps) as progress_bar:
         for i, t in enumerate(timesteps):