Update app.py

app.py

@@ -1,3 +1,18 @@
+import math
+from collections import OrderedDict
+import sys
+import torch
+import torch.nn as nn
+import torch.nn.functional as F
+import torch.optim as optim
+from torchvision import datasets, transforms
+import torchvision
+import numpy as np
+import matplotlib.pyplot as plt
+import pandas as pd
+import seaborn as sn
+from torch_lr_finder import LRFinder
+from torch.optim.lr_scheduler import OneCycleLR
 import torch, torchvision
 from torchvision import transforms
 import numpy as np
@@ -12,6 +27,7 @@ from pytorch_lightning.callbacks import LearningRateMonitor
 from pytorch_lightning.callbacks.progress import TQDMProgressBar
 from pytorch_lightning.loggers import CSVLogger
 from pytorch_lightning.loggers import TensorBoardLogger
+from torchmetrics import Accuracy
 
 class LitResnet(LightningModule):
     def __init__(self, num_classes=10, lr=0.05):