dreamdrop-art commited on
Commit
aa28d86
1 Parent(s): b912fed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -519,10 +519,9 @@ with gr.Blocks(css=css) as demo:
519
  with gr.Tab("Visual Question Answering"):
520
  gr.load("models/dandelin/vilt-b32-finetuned-vqa", title="ViLT VQA")
521
 
522
- model_cache = gr.Textbox(interactive=False, visible=False)
523
- hf_model_cache = gr.Textbox(interactive=False, visible=False)
524
- model.change(change_checkpoint, inputs=model, outputs=model_cache)
525
- hf_model.change(change_checkpoint, inputs=hf_model, outputs=hf_model_cache)
526
 
527
 
528
  demo.queue(max_size=80, api_open=False).launch(max_threads=256, show_api=False)
 
519
  with gr.Tab("Visual Question Answering"):
520
  gr.load("models/dandelin/vilt-b32-finetuned-vqa", title="ViLT VQA")
521
 
522
+
523
+ model.change(change_checkpoint, inputs=model, outputs=model)
524
+ hf_model.change(change_checkpoint, inputs=hf_model, outputs=hf_model)
 
525
 
526
 
527
  demo.queue(max_size=80, api_open=False).launch(max_threads=256, show_api=False)