Spaces:
Runtime error
Runtime error
Upload luciddreamer.py with huggingface_hub
Browse files- luciddreamer.py +2 -2
luciddreamer.py
CHANGED
@@ -213,7 +213,7 @@ class LucidDreamer:
|
|
213 |
except OSError as e:
|
214 |
print("Error: %s - %s." % (e.filename, e.strerror))
|
215 |
|
216 |
-
def render_video(self, preset, example_name=None):
|
217 |
if example_name and example_name != 'DON\'T':
|
218 |
videopath = os.path.join('examples', f'{example_name}_{preset}.mp4')
|
219 |
depthpath = os.path.join('examples', f'depth_{example_name}_{preset}.mp4')
|
@@ -255,7 +255,7 @@ class LucidDreamer:
|
|
255 |
imageio.mimwrite(depthpath, depthlist, fps=60, quality=8)
|
256 |
return videopath, depthpath
|
257 |
|
258 |
-
def training(self):
|
259 |
if not self.scene:
|
260 |
raise('Build 3D Scene First!')
|
261 |
|
|
|
213 |
except OSError as e:
|
214 |
print("Error: %s - %s." % (e.filename, e.strerror))
|
215 |
|
216 |
+
def render_video(self, preset, example_name=None, progress=gr.Progress()):
|
217 |
if example_name and example_name != 'DON\'T':
|
218 |
videopath = os.path.join('examples', f'{example_name}_{preset}.mp4')
|
219 |
depthpath = os.path.join('examples', f'depth_{example_name}_{preset}.mp4')
|
|
|
255 |
imageio.mimwrite(depthpath, depthlist, fps=60, quality=8)
|
256 |
return videopath, depthpath
|
257 |
|
258 |
+
def training(self, progress=gr.Progress()):
|
259 |
if not self.scene:
|
260 |
raise('Build 3D Scene First!')
|
261 |
|