Update app.py

app.py

@@ -318,7 +318,7 @@ class ForgeryDetector:
         
         # Run localization
         with torch.no_grad():
-            logits, decoder_features = self.model(image_tensor)
+            logits, _,decoder_features = self.model(image_tensor)
             prob_map = torch.sigmoid(logits).cpu().numpy()[0, 0]
         
         # Resize probability map to match original image size to avoid index mismatch errors