while循环语句不结束，如何修改才能正常运行(语言-java|开发工具-intellij-idea)

Question

这是道编程练习题，目的写一个ATM机，开始预设账户余额900 ，用户输入金额小于或等于900则输出“剩余金额：900，是否继续（'1'：结束，'2'：继续）：” 然后用户输入1 就输出“取款结束”，然后整个程序的循环结束。
我这里遇到的问题是 这里的while循环不结束，在IDEA上是这样显示的

在一个线上做题平台上的编译器是这样显示的

我是实在不会改，请出出手斧正。

//package step4;

import java.util.Scanner;

public class ForPractice4 {
    public static void main(String[] args) {
        System.out.println("欢迎使用中国人民银行ATM取款机");
        ATM teller = new ATM();
        int i = 0;
        while (i == 0 || i ==2) {
            int cash = teller.input();
             i = teller.teller(cash);

        }
    }
}

class ATM {
    Scanner sc = new Scanner(System.in);
    public int input() {

        System.out.print("输入取款金额:");

        return sc.nextInt();
    }

    public int teller(int cash) {
        int balance = 900;
        int choose = sc.nextInt();
        if (cash <= balance) {
            System.out.print("剩余金额：900，是否继续（'1'：结束，'2'：继续）：");

            switch (choose) {

                case 1:
                    System.out.println("取款结束");
                    return 1;

                case 2:
                    return 2;
            }
        } else {
            System.out.println("目前余额:" + balance + "无法满足您的取款需求！");

        }
        return 0;
    }
}

Answer 1

稍等~