Spaces:

ttd22
/

segment-anything-model

Build error

Trang Dang commited on Apr 29, 2024

Commit

99ab932

•

1 Parent(s): e01d5f6

change color

Files changed (1) hide show

app.py CHANGED Viewed

@@ -46,7 +46,7 @@ def server(input: Inputs, output: Outputs, session: Session):
             prob, prediction = run.pred(src)
             fig, axes = plt.subplots(1, 2, figsize=(60, 30))
-            im = axes[0].imshow(prob, cmap='turbo')
             axes[0].set_title("Probability Map")
             cbar = fig.colorbar(im, ax=axes[0])

             prob, prediction = run.pred(src)
             fig, axes = plt.subplots(1, 2, figsize=(60, 30))
+            im = axes[0].imshow(prob, cmap='viridis')
             axes[0].set_title("Probability Map")
             cbar = fig.colorbar(im, ax=axes[0])