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 |
|