Update app.py

app.py

@@ -8,6 +8,12 @@ model.load_state_dict(torch.load("model.pth", map_location=torch.device('cpu')))
 model.eval()
 device = torch.device("cpu")
 
+transform = transforms.Compose([
+    transforms.Resize((224, 224)),
+    transforms.ToTensor(),
+    transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
+])
+
 import gradio as gr
 from PIL import Image