change color

app.py

@@ -34,7 +34,7 @@ def server(input: Inputs, output: Outputs, session: Session):
     def image():
         if input.image_input():
             src = input.image_input()[0]['datapath']
-            img = {"src": src, "width": "250px"} 
+            img = {"src": src, "width": "500px"} 
             return img
         return None
 
@@ -46,8 +46,10 @@ def server(input: Inputs, output: Outputs, session: Session):
             prob, prediction = run.pred(src)
             fig, axes = plt.subplots(1, 2, figsize=(60, 30))
 
-            axes[0].imshow(prob, cmap='turbo')
+            im = axes[0].imshow(prob, cmap='turbo')
             axes[0].set_title("Probability Map")
+            cbar = fig.colorbar(im, ax=axes[0])
+            
 
             axes[1].imshow(prediction, cmap='gray') 
             axes[1].set_title("Prediction")