Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -321,7 +321,7 @@ with gr.Blocks(css=css, js=js) as demo:
|
|
321 |
cache_examples=False
|
322 |
)
|
323 |
predict_button.click(
|
324 |
-
|
325 |
inputs=[image_input, gen_threshold, char_threshold, output_mode],
|
326 |
outputs=[output_tags, output_visualization]
|
327 |
)
|
|
|
321 |
cache_examples=False
|
322 |
)
|
323 |
predict_button.click(
|
324 |
+
tagger.predict_on_gpu, # Call class method
|
325 |
inputs=[image_input, gen_threshold, char_threshold, output_mode],
|
326 |
outputs=[output_tags, output_visualization]
|
327 |
)
|