Update app.py

app.py

@@ -52,11 +52,8 @@ def load_model():
 
 # ── Prediction ─────────────────────────────
 def predict(model, class_indices, img, input_shape):
-    # Use model's expected input size
-    if input_shape and len(input_shape) >= 3:
-        target_size = (input_shape[1], input_shape[2])  # (height, width)
-    else:
-        target_size = (160, 160)  # fallback
+
+    target_size = (160, 160)  # fallback
 
     img = img.convert("RGB").resize(target_size)
     arr = np.array(img, dtype=np.float32) / 255.0