neverix commited on
Commit
fb4636f
1 Parent(s): 17d9fef

MaybeMaybeMaybe

Browse files
Files changed (3) hide show
  1. app.py +5 -1
  2. packages.txt +2 -1
  3. requirements.txt +2 -1
app.py CHANGED
@@ -108,7 +108,9 @@ def render(mesh, mat):
108
 
109
  def main():
110
  os.environ["PYOPENGL_PLATFORM"] = "egl"
111
- os.environ["DISPLAY"] = "0:0"
 
 
112
 
113
  midas = MidasDepth()
114
  def fn(pic, *args):
@@ -132,6 +134,8 @@ def main():
132
  ], title="DALL·E 6D", description="Lift DALL·E 2 (or any other model) into 3D!")
133
  gr.TabbedInterface([interface], ["Warp 3D images"]).launch()
134
 
 
 
135
 
136
  if __name__ == '__main__':
137
  main()
108
 
109
  def main():
110
  os.environ["PYOPENGL_PLATFORM"] = "egl"
111
+ from pyvirtualdisplay import Display
112
+ disp = Display()
113
+ disp.start()
114
 
115
  midas = MidasDepth()
116
  def fn(pic, *args):
134
  ], title="DALL·E 6D", description="Lift DALL·E 2 (or any other model) into 3D!")
135
  gr.TabbedInterface([interface], ["Warp 3D images"]).launch()
136
 
137
+ disp.stop()
138
+
139
 
140
  if __name__ == '__main__':
141
  main()
packages.txt CHANGED
@@ -1,3 +1,4 @@
1
  mesa-utils
2
  libglfw3-dev
3
- libgles2-mesa-dev
 
1
  mesa-utils
2
  libglfw3-dev
3
+ libgles2-mesa-dev
4
+ xvfb
requirements.txt CHANGED
@@ -6,4 +6,5 @@ pyrender
6
  torchvision==0.12.0
7
  timm==0.6.5
8
  opencv-python==4.4.0.44
9
- # git+https://github.com/mmatl/pyopengl.git
 
6
  torchvision==0.12.0
7
  timm==0.6.5
8
  opencv-python==4.4.0.44
9
+ # git+https://github.com/mmatl/pyopengl.git
10
+ pyvirtualdisplay