From ef58d6048200902c4e6b97e9fdab6f2d44c96894 Mon Sep 17 00:00:00 2001
From: Anjok07 <68268275+Anjok07@users.noreply.github.com>
Date: Sun, 17 Apr 2022 15:39:50 -0500
Subject: [PATCH] Update inference_v5_ensemble.py

---
 inference_v5_ensemble.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/inference_v5_ensemble.py b/inference_v5_ensemble.py
index 2b410cd..2d29a8e 100644
--- a/inference_v5_ensemble.py
+++ b/inference_v5_ensemble.py
@@ -501,6 +501,14 @@ def main(window: tk.Wm, text_widget: tk.Text, button_widget: tk.Button, progress
                                 _, bin_image = cv2.imencode('.jpg', image)
                                 bin_image.tofile(f)
                                 
+                        text_widget.write(base_text + 'Clearing CUDA Cache... ') 
+                                
+                        torch.cuda.empty_cache()
+                        time.sleep(3)     
+                        
+                        text_widget.write('Done!\n')
+                                
+                              
                         text_widget.write(base_text + 'Completed Seperation!\n\n')  
                         
                 # Emsembling Outputs