Spaces:
Running
on
A100
Running
on
A100
Update app.py
Browse files
app.py
CHANGED
@@ -224,6 +224,7 @@ with gr.Blocks(theme=gr.themes.Soft(), analytics_enabled=False, title="Keyframe
|
|
224 |
out_json = gr.JSON(label="Parsed JSON")
|
225 |
ok_flag = gr.Checkbox(label="Valid JSON", value=False, interactive=False)
|
226 |
|
|
|
227 |
def on_submit(img):
|
228 |
text, js, ok = generate_json(img)
|
229 |
return text, js, ok
|
|
|
224 |
out_json = gr.JSON(label="Parsed JSON")
|
225 |
ok_flag = gr.Checkbox(label="Valid JSON", value=False, interactive=False)
|
226 |
|
227 |
+
@spaces.GPU # ensures a GPU task is registered
|
228 |
def on_submit(img):
|
229 |
text, js, ok = generate_json(img)
|
230 |
return text, js, ok
|