Fix model loading

app.py

@@ -4,12 +4,13 @@ from fastai.vision.all import *
 # Load model
 classes = ['SUV', 'bus', 'cabriolet', 'crossover', 'hatchback', 'limousine', 'pickup', 'sedan', 'sports', 'truck', 'van']
 
-dls = ImageDataLoaders.from_folder(
-    Path("."),  # dummy path
-    valid_pct=0.0,
+dblock = DataBlock(
+    blocks=(ImageBlock, CategoryBlock(vocab=classes)),
+    get_items=get_image_files,
+    splitter=RandomSplitter(0.2),
     item_tfms=Resize(224)
 )
-dls.vocab = classes
+dls = dblock.dataloaders(Path("."), bs=1)
 
 learn = cnn_learner(dls, resnet18)
 learn.load("model.pth")