Update app.py

app.py

@@ -5,6 +5,7 @@ import torch
 import torch.nn.functional as F
 import torchvision.transforms as transforms
 import torchvision
+from torchkeras import plots 
 import numpy as np
 import yaml
 from huggingface_hub import hf_hub_download