Spaces:
Running
on
Zero
Running
on
Zero
Ava Pun
commited on
Commit
·
6e12b3d
1
Parent(s):
bce7f3d
add spaces.gpu
Browse files
app.py
CHANGED
@@ -68,7 +68,7 @@ def main():
|
|
68 |
|
69 |
# Define Gradio interface
|
70 |
demo = gr.Interface(
|
71 |
-
fn=generator.
|
72 |
title='BrickGPT Demo',
|
73 |
description='Official demo for [BrickGPT](https://avalovelace1.github.io/BrickGPT/), the first approach for generating physically stable toy brick models from text prompts.\n\n'
|
74 |
'The model is restricted to creating structures made of 1-unit-tall cuboid bricks on a 20x20x20 grid. It was trained on a dataset of 21 object categories: '
|
@@ -94,7 +94,7 @@ def main():
|
|
94 |
run_on_click=True,
|
95 |
)
|
96 |
|
97 |
-
concurrency_limit =
|
98 |
demo.queue(default_concurrency_limit=concurrency_limit)
|
99 |
demo.launch(share=True)
|
100 |
|
|
|
68 |
|
69 |
# Define Gradio interface
|
70 |
demo = gr.Interface(
|
71 |
+
fn=generator.generate_bricks,
|
72 |
title='BrickGPT Demo',
|
73 |
description='Official demo for [BrickGPT](https://avalovelace1.github.io/BrickGPT/), the first approach for generating physically stable toy brick models from text prompts.\n\n'
|
74 |
'The model is restricted to creating structures made of 1-unit-tall cuboid bricks on a 20x20x20 grid. It was trained on a dataset of 21 object categories: '
|
|
|
94 |
run_on_click=True,
|
95 |
)
|
96 |
|
97 |
+
concurrency_limit = 1 if os.environ.get('CONCURRENCY_LIMIT') is None else int(os.environ.get('CONCURRENCY_LIMIT'))
|
98 |
demo.queue(default_concurrency_limit=concurrency_limit)
|
99 |
demo.launch(share=True)
|
100 |
|