Update app.py

app.py

@@ -17,7 +17,7 @@ labels = requests.get(LABELS_URL).text.strip().split("\n")
 
 # Image transform
 transform = transforms.Compose([
-    transforms.Resize((224, 224)),
+    transforms.Resize((224, 224)),  # Ensuring image is resized to (224, 224)
     transforms.ToTensor(),
 ])
 
@@ -62,7 +62,7 @@ def main():
 
     def load_image():
         if uploaded_file:
-            return Image.open(uploaded_file).resize(244,244)
+            return Image.open(uploaded_file).resize((224, 224))  # Resize to (224, 224)
         elif selected_preset:
             return Image.open(presets[selected_preset])
         else:
@@ -93,7 +93,7 @@ def main():
             st.markdown("### Adversarial Prediction")
             st.error(adversarial_label)
             # Ensure the perturbed image has the same size as the original image
-            perturbed_image_resized = perturbed_image.resize(image.size)  # Resize to match original image size
+            perturbed_image_resized = perturbed_image.resize((224, 224))  # Resize to (224, 224)
             st.image(perturbed_image_resized, caption="Perturbed Image", use_container_width=True)
     else:
         st.warning("Please upload image.")