neverix commited on
Commit
be529c0
·
1 Parent(s): 419de6a
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -92,7 +92,7 @@ def args_to_mat(tx, ty, tz, rx, ry, rz):
92
 
93
 
94
  def render(mesh, mat):
95
- mesh = pyrender.mesh.Mesh.from_trimesh(mesh)
96
  scene = pyrender.Scene(ambient_light=np.array([1.0, 1.0, 1.0]))
97
  camera = pyrender.PerspectiveCamera(yfov=np.pi / 2, aspectRatio=1.0)
98
  scene.add(camera, pose=mat)
@@ -130,8 +130,9 @@ def main():
130
  gr.outputs.Image(type="numpy", label="result"),
131
  gr.outputs.Image(type="numpy", label="depth"),
132
  gr.outputs.Video(label="interpolated")
133
- ], title="DALL·E 6D", description="Lift DALL·E 2 (or any other model) into 3D!")
134
- gr.TabbedInterface([interface], ["Warp 3D images"]).launch()
 
135
 
136
 
137
  if __name__ == '__main__':
 
92
 
93
 
94
  def render(mesh, mat):
95
+ mesh = pyrender.mesh.Mesh.from_trimesh(mesh, smooth=False)
96
  scene = pyrender.Scene(ambient_light=np.array([1.0, 1.0, 1.0]))
97
  camera = pyrender.PerspectiveCamera(yfov=np.pi / 2, aspectRatio=1.0)
98
  scene.add(camera, pose=mat)
 
130
  gr.outputs.Image(type="numpy", label="result"),
131
  gr.outputs.Image(type="numpy", label="depth"),
132
  gr.outputs.Video(label="interpolated")
133
+ ])
134
+ gr.TabbedInterface([interface], ["Warp 3D images"],
135
+ title="DALL·E 6D", description="Lift DALL·E 2 (or any other model) into 3D!").launch()
136
 
137
 
138
  if __name__ == '__main__':