Fix model loading

app.py

@@ -2,6 +2,15 @@ import gradio as gr
 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,
+    item_tfms=Resize(224)
+)
+dls.vocab = classes
+
 learn = cnn_learner(dls, resnet18)
 learn.load("model.pth")