Update app.py

app.py

@@ -6,9 +6,17 @@ from PIL import Image
 # Load your PyTorch model
 model = torch.load("model.pth",map_location=torch.device('cpu'))
 
+classes = ['bom', 'ruim']
+
+def preprocess(image):
+    image = image.resize((224, 224))
+    image_tensor = ToTensor()(image)
+    image_tensor = image_tensor.unsqueeze(0)
+    return image_tensor
+
 # Define the function for image classification
 def classify_image(image):
-    image_tensor = ToTensor()(image).unsqueeze(0)
+    image_tensor = preprocess(image)
 
     # Perform inference using your PyTorch model
     with torch.no_grad():