def after_generation(self):
        """Call this function after running a generation task"""
        self.img.setEnabled(True)
        self.generate.setEnabled(True)
        self.browse_results.setEnabled(True)
        self.parent.store_dimension_settings()
Call this function after running a generation task
src/frontend/gui/base_widget.py