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_processn(outputs, img_size)
+    processed_outputs = feature_extractor.post_process(outputs, img_size)
     print(outputs.keys())
     return (
         processed_outputs[0],
@@ -108,7 +108,7 @@ with gr.Blocks(css=css) as app:
                             pathlib.Path("cd45rb_test_imgs").rglob("*_HE.png")
                         )
                     ],
-                    samples_per_page=3,
+                    samples_per_page=2,
                 )
             with gr.Row():
                 with gr.Column():