Spaces:
Running
on
A100
Running
on
A100
Commit
·
60cc681
1
Parent(s):
74ccc45
Revert "debugging ldraw path again"
Browse filesThis reverts commit c229f95bbda11465703810b7ae3ecb5d767ac2f1.
- app.py +1 -2
- render_lego.py +0 -10
app.py
CHANGED
@@ -52,8 +52,7 @@ def main():
|
|
52 |
|
53 |
# Generate LEGO
|
54 |
print(f'Generating LEGO for prompt: "{prompt}"')
|
55 |
-
|
56 |
-
output = '2x4 (0,0,0)\n'
|
57 |
print(f'Finished generation!')
|
58 |
|
59 |
# Render results and write to files
|
|
|
52 |
|
53 |
# Generate LEGO
|
54 |
print(f'Generating LEGO for prompt: "{prompt}"')
|
55 |
+
output = model(prompt)
|
|
|
56 |
print(f'Finished generation!')
|
57 |
|
58 |
# Render results and write to files
|
render_lego.py
CHANGED
@@ -1,14 +1,4 @@
|
|
1 |
-
import os
|
2 |
-
from pathlib import Path
|
3 |
-
|
4 |
from legogpt.render_lego import main as render_lego_main
|
5 |
|
6 |
if __name__ == '__main__':
|
7 |
-
ldraw_lib_path = os.environ.get('LDRAW_LIBRARY_PATH')
|
8 |
-
if not ldraw_lib_path or not os.path.exists(ldraw_lib_path):
|
9 |
-
# Default path to LDraw library is home directory
|
10 |
-
ldraw_lib_path = Path.home() / 'ldraw'
|
11 |
-
ldraw_lib_path = os.path.abspath(ldraw_lib_path)
|
12 |
-
print('LDRAW lib path:', ldraw_lib_path)
|
13 |
-
|
14 |
render_lego_main()
|
|
|
|
|
|
|
|
|
1 |
from legogpt.render_lego import main as render_lego_main
|
2 |
|
3 |
if __name__ == '__main__':
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
4 |
render_lego_main()
|