Fix model loading

app.py

@@ -2,18 +2,7 @@ import gradio as gr
 from fastai.vision.all import *
 
 # Load model
-classes = ['SUV', 'bus', 'cabriolet', 'crossover', 'hatchback', 'limousine', 'pickup', 'sedan', 'sports', 'truck', 'van']
-
-dblock = DataBlock(
-    blocks=(ImageBlock, CategoryBlock(vocab=classes)),
-    get_items=get_image_files,
-    splitter=RandomSplitter(0.2),
-    item_tfms=Resize(224)
-)
-dls = dblock.dataloaders(Path("."), bs=1)
-
-learn = cnn_learner(dls, resnet18)
-learn.load("model.pth")
+learn = load_learner("model.pkl")
 
 # Inference function
 def classify_image(img):