reload

app.py

@@ -1,6 +1,5 @@
 from pathlib import Path
 from typing import List, Dict, Tuple
-import matplotlib.colors as mpl_colors
 import pandas as pd
 import seaborn as sns
 import shinyswatch
@@ -31,7 +30,6 @@ def server(input: Inputs, output: Outputs, session: Session):
     @render.image
     def image():
         if input.image_input():
-            # print(input.image_input())
             src = input.image_input()[0]['datapath']
             img = {"src": src, "width": "500px"} 
             return img
@@ -42,13 +40,13 @@ def server(input: Inputs, output: Outputs, session: Session):
     def plot_output():
         if input.image_input():
             src = input.image_input()[0]['datapath']
-            single_patch_prob, single_patch_prediction = run.pred(src)
+            prob, prediction = run.pred(src)
             fig, axes = plt.subplots(1, 2, figsize=(15, 5))
 
-            axes[0].imshow(single_patch_prob, cmap='gray')
+            axes[0].imshow(prob, cmap='gray')
             axes[0].set_title("Probability Map")
 
-            im = axes[1].imshow(single_patch_prediction) 
+            im = axes[1].imshow(prediction) 
             axes[1].set_title("Prediction")
             cbar = fig.colorbar(im, ax=axes[1])