Update app.py

app.py

@@ -54,7 +54,7 @@ def plot_ui():
 
 @module.server
 def plot_server(input, output, session, r, opacity):
-    @render.ui
+    @render.plot
     def plot_prediction():
         fig, ax = plt.subplots()