add model transforms

app.py

@@ -1,10 +1,10 @@
 import gradio as gr
 import timm
 import torch
-
+from torchvision import transforms
 model = timm.create_model("mobileone_s2", pretrained = False)
 model.head.fc = torch.nn.Linear(model.head.fc.in_features,3)
-transforms = timm.data.create_transform(**timm.data.resolve_data_config(model.pretrained_cfg)).transforms
+transforms = transforms.Compose(timm.data.create_transform(**timm.data.resolve_data_config(model.pretrained_cfg)).transforms)
 model.load_state_dict(torch.load("olive-classifier.pth", map_location=torch.device('cpu'), weights_only=True))
 model.eval()