liuhuadai commited on
Commit
3109a31
1 Parent(s): 42b6dae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -101,7 +101,7 @@ class GenSamples:
101
  record_dicts.append(record_dict)
102
  return record_dicts
103
 
104
-
105
  def infer(ori_prompt):
106
 
107
  prompt = dict(ori_caption=ori_prompt,struct_caption=f'<{ori_prompt}& all>')
 
101
  record_dicts.append(record_dict)
102
  return record_dicts
103
 
104
+ @spaces.GPU(enable_queue=True)
105
  def infer(ori_prompt):
106
 
107
  prompt = dict(ori_caption=ori_prompt,struct_caption=f'<{ori_prompt}& all>')