update app.py

app.py

@@ -9,7 +9,6 @@ import torch.nn as nn
 import torch.nn.functional as F
 import torchvision
 import matplotlib.pyplot as plt
-import torch.nn.functional as F
 from pl_bolts.datamodules import CIFAR10DataModule
 from pl_bolts.transforms.dataset_normalizations import cifar10_normalization
 from pytorch_lightning import LightningModule, Trainer, seed_everything
@@ -40,7 +39,7 @@ device = torch.device("cpu")
 classes = ('plane', 'car', 'bird', 'cat', 'deer',
            'dog', 'frog', 'horse', 'ship', 'truck')
 
-model = LitResnet(lr=0.05).load_from_checkpoint("/content/weights.ckpt")
+model = LitResnet(lr=0.05).load_from_checkpoint("weights_92.ckpt")
 
 device = torch.device("cpu")