huntrezz commited on
Commit
2505cbf
1 Parent(s): 7e5717f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -67,7 +67,8 @@ def plot_depth_map(depth_map, original_image):
67
 
68
  ax.view_init(elev=150, azim=90, roll=40)
69
  plt.axis('off')
70
- plt.close(fig)
 
71
 
72
  fig.canvas.draw()
73
  img = np.frombuffer(fig.canvas.tostring_rgb(), dtype=np.uint8)
 
67
 
68
  ax.view_init(elev=150, azim=90, roll=40)
69
  plt.axis('off')
70
+
71
+ plt.show()
72
 
73
  fig.canvas.draw()
74
  img = np.frombuffer(fig.canvas.tostring_rgb(), dtype=np.uint8)