Spaces:
Running
on
Zero
Running
on
Zero
znchen
commited on
Commit
•
bd9c628
1
Parent(s):
48a06e6
Update app.py
Browse files
app.py
CHANGED
|
@@ -41,6 +41,7 @@ def get_box_inputs(prompts):
|
|
| 41 |
box_inputs.append((prompt[0], prompt[1], prompt[3], prompt[4]))
|
| 42 |
return box_inputs
|
| 43 |
|
|
|
|
| 44 |
def rag_gen(
|
| 45 |
box_prompt_image,
|
| 46 |
prompt,
|
|
|
|
| 41 |
box_inputs.append((prompt[0], prompt[1], prompt[3], prompt[4]))
|
| 42 |
return box_inputs
|
| 43 |
|
| 44 |
+
@spaces.GPU
|
| 45 |
def rag_gen(
|
| 46 |
box_prompt_image,
|
| 47 |
prompt,
|