Transferred to a CPU model.

app.py

@@ -15,9 +15,8 @@ np.set_printoptions(threshold=sys.maxsize)
 # %% app.ipynb 4
 from lenet import LeNet5
 # Allowlist the custom class
-MODEL_PATH = Path("models/lenet5.pt")
+MODEL_PATH = Path("models/lenet5-cpu.pt")
 model = torch.load(MODEL_PATH, weights_only=False)
-model = model.to('cpu')  # Move model to CPU
 model.eval()
 
 def predict(img):