Update app.py

app.py

@@ -81,6 +81,9 @@ def server(input: Inputs, output: Outputs, session: Session):
             current_mask = prediction.get_sidewalk_prediction(image, model, processor, device)
             logger.info("Inference completed.")
 
+            # Trigger the plot update
+            output.plot_image_and_mask.invalidate()
+
     @output
     @render.plot
     def plot_image_and_mask():