neverix commited on
Commit
7250396
1 Parent(s): 61a9c25
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -102,14 +102,12 @@ def render(mesh, mat):
102
  mask = d == 0
103
  rgb = rgb.copy()
104
  rgb[mask] = 0
105
- res = Image.fromarray(np.concatenate((rgb,
106
- ((mask[..., np.newaxis]) == 0)
107
- .astype(np.uint8) * 255), axis=-1))
108
  return res
109
 
110
 
111
  def main():
112
- os.environ["PYOPENGL_PLATFORM"] = "osmesa"
113
 
114
  midas = MidasDepth()
115
  def fn(pic, *args):
 
102
  mask = d == 0
103
  rgb = rgb.copy()
104
  rgb[mask] = 0
105
+ res = Image.fromarray(np.concatenate((rgb, ((mask[..., np.newaxis]) == 0).astype(np.uint8) * 255), axis=-1))
 
 
106
  return res
107
 
108
 
109
  def main():
110
+ # os.environ["PYOPENGL_PLATFORM"] = "osmesa"
111
 
112
  midas = MidasDepth()
113
  def fn(pic, *args):