Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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>')
|