Update network.py

network.py

@@ -1,3 +1,29 @@
+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
+from torch_lr_finder import LRFinder
+from torch.optim.lr_scheduler import OneCycleLR
+import torch, torchvision
+from torchvision import transforms
+import numpy as np
+import gradio as gr
+from PIL import Image
+from pytorch_grad_cam import GradCAM
+from pytorch_grad_cam.utils.image import show_cam_on_image
+import gradio as gr
+from pytorch_lightning import LightningModule, Trainer, seed_everything
+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
+from models import custom_resnet
+
 class LitResnet(LightningModule):
     def __init__(self, num_classes=10, lr=0.05):
         super().__init__()