diff --git a/vast_start.py b/vast_start.py index afb0379..c1a709a 100644 --- a/vast_start.py +++ b/vast_start.py @@ -247,10 +247,12 @@ def main(): except Exception as e: print(f"\nAn error occurred during instance setup: {e}") finally: - # Ask user if they want to clean up, or just do it automatically - # Based on original script, it immediately destroys the instance after the subprocess ends - print("\nCleaning up...") if instance_id: + try: + input("\nPress ENTER to destroy the instance and clean up...") + except (EOFError, KeyboardInterrupt): + pass + print("\nCleaning up...") destroy_instance(instance_id) if __name__ == "__main__":