Spaces:

AIGC-Audio
/

AudioLCM

Runtime error

liuhuadai commited on Jun 5, 2024

Commit

3109a31

verified ·

1 Parent(s): 42b6dae

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -101,7 +101,7 @@ class GenSamples:
                 record_dicts.append(record_dict)
         return record_dicts
 def infer(ori_prompt):
     prompt = dict(ori_caption=ori_prompt,struct_caption=f'<{ori_prompt}& all>')

                 record_dicts.append(record_dict)
         return record_dicts
+@spaces.GPU(enable_queue=True)
 def infer(ori_prompt):
     prompt = dict(ori_caption=ori_prompt,struct_caption=f'<{ori_prompt}& all>')