Update app.py

app.py

@@ -16,9 +16,11 @@ from torch.nn import functional as F
 import matplotlib.pyplot as plt
 import gradio as gr
 from PIL import Image
-from Dataset.testalbumentation import TestAlbumentation
-from Model.Lit_cifar_module import LitCifar
 from utils import *
+from datasets import cifar10_dataset
+from models.CUSTOMRESNET import Model
+from utils.incorrect_images import incorrect
+
 
 model = LitCifar().cpu()
 model.load_state_dict(torch.load('model_weights.pth', map_location=torch.device('cpu')))