Update custom_resnet.py

custom_resnet.py

@@ -2,18 +2,10 @@ from pytorch_lightning import LightningModule, Trainer, seed_everything
 import torch
 import pandas as pd
 import numpy as np
-import gradio as gr
-from PIL import Image
+import torch.nn as nn
 from torch.nn import functional as F
-from collections import OrderedDict
 from torchvision import transforms
-from pytorch_grad_cam import GradCAM
-from pytorch_grad_cam.utils.image import show_cam_on_image
-from pytorch_grad_cam.utils.model_targets import ClassifierOutputTarget
 from pytorch_lightning import LightningModule, Trainer, seed_everything
-import albumentations as A
-from albumentations.pytorch import ToTensorV2
-import torchvision.transforms as T
 
 class LitResnet(LightningModule):
     def __init__(self, lr=0.01, drop= 0.05, norm='BN',groupsize=1):