Update app.py

app.py

@@ -222,7 +222,7 @@ transform = T.Compose([
 
 def predict(image):
     # Preprocess the image
-    # img_tensor = transform(image).unsqueeze(0)  # Add batch dimension
+    img_tensor = transform(image).unsqueeze(0)  # Add batch dimension
     
     # # Make prediction
     # with torch.no_grad():