Update app.py

app.py

@@ -269,6 +269,17 @@ css = """
 }
 """
 
+def change_checkpoint(name, progress=gr.Progress()):
+    progress(0, desc="Starting")
+    time.sleep(0.5)
+    progress(0.25, desc="Changing")
+    time.sleep(1)
+    progress(0.95, desc="Loading to GPU")
+    time.sleep(0.5)
+    progress(0.100, desc="Ready")
+    return name
+    
+
 with gr.Blocks(css=css) as demo:
     with gr.Row():
         with gr.Column(scale=6):
@@ -507,6 +518,11 @@ with gr.Blocks(css=css) as demo:
         #    gr.load("models/mattmdjaga/segformer_b2_clothes", title="SegFormer Segmentation")
         with gr.Tab("Visual Question Answering"):
             gr.load("models/dandelin/vilt-b32-finetuned-vqa", title="ViLT VQA")
+
+        model_cache = gr.Textbox(interactive=False, visible=False)
+        hf_model_cache = gr.Textbox(interactive=False, visible=False)
+        model.change(change_checkpoint, inputs=model, outputs=model_cache)
+        hf_model.change(change_checkpoint, inputs=hf_model, outputs=hf_model_cache)
+        
         
-            
 demo.queue(max_size=80, api_open=False).launch(max_threads=256, show_api=False)