Update app.py

app.py

@@ -16,7 +16,7 @@ def make_prediction(img, feature_extractor, model):
     inputs = feature_extractor(img, return_tensors="pt")
     outputs = model(**inputs)
     img_size = torch.tensor([tuple(reversed(img.size))])
-    processed_outputs = feature_extractor.post_process_object_detection(outputs, img_size)
+    processed_outputs = feature_extractor.post_processn(outputs, img_size)
     print(outputs.keys())
     return (
         processed_outputs[0],