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