change

app.py

@@ -46,10 +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)
+            axes[0].imshow(prob, cmap='turbo')
             axes[0].set_title("Probability Map")
 
-            im = axes[1].imshow(prediction, cmap='gray') 
+            axes[1].imshow(prediction, cmap='gray') 
             axes[1].set_title("Prediction")
 
             for ax in axes: