space

app.py

@@ -4,6 +4,7 @@ import demo_tasks
 from functools import partial
 from data.prefix_instruction import get_layout_instruction
 from visualcloze import VisualClozeModel
+import spaces
 
 
 max_grid_h = 5
@@ -36,7 +37,7 @@ If you find VisualCloze is helpful, please consider to star ⭐ the <a href='htt
 If our work is useful for your research, please consider citing:
 ```bibtex
 @article{li2025visualcloze,
-  title={VisualCloze : A Universal Image Generation Framework via Visual In-Context Learning},
+  title={VisualCloze: A Universal Image Generation Framework via Visual In-Context Learning},
   author={Li, Zhong-Yu and Du, ruoyi and Yan, Juncheng and Zhuo, Le and Li, Zhen and Gao, Peng and Ma, Zhanyu and Cheng, Ming-Ming},
   booktitle={arXiv preprint arxiv:},
   year={2025}
@@ -304,6 +305,7 @@ def create_demo(model):
             updates.append(gr.update(elem_id="layout_prompt", value=get_layout_instruction(w, actual_h)))
             return updates
 
+        @spaces.GPU
         def generate_image(*inputs):
             images = []
             for i in range(model.grid_h):