Ava Pun commited on
Commit
6e12b3d
·
1 Parent(s): bce7f3d

add spaces.gpu

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -68,7 +68,7 @@ def main():
68
 
69
  # Define Gradio interface
70
  demo = gr.Interface(
71
- fn=generator.generate_bricks_subprocess,
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 = 2 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
 
 
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