Final test

app_caption.py

@@ -37,7 +37,8 @@ def create_demo() -> gr.Blocks:
         gr.Examples(examples=examples,
                     inputs=inputs,
                     outputs=outputs,
-                    fn=model.run_caption)
+                    fn=model.run_caption,
+                    cache_examples=os.getenv('SYSTEM') == 'spaces')
 
         run_button.click(fn=model.run_caption, inputs=inputs, outputs=outputs)
     return demo

app_vqa.py

@@ -42,7 +42,8 @@ def create_demo() -> gr.Blocks:
         gr.Examples(examples=examples,
                     inputs=inputs,
                     outputs=outputs,
-                    fn=model.run_vqa)
+                    fn=model.run_vqa,
+                    cache_examples=os.getenv('SYSTEM') == 'spaces')
 
         run_button.click(fn=model.run_vqa, inputs=inputs, outputs=outputs)
     return demo

prismer_model.py

@@ -34,10 +34,10 @@ def download_models() -> None:
         subprocess.run(shlex.split('python download_checkpoints.py --download_experts=True'), cwd='prismer')
 
     model_names = [
-        # 'vqa_prismer_base',
-        # 'vqa_prismer_large',
+        'vqa_prismer_base',
+        'vqa_prismer_large',
         'pretrain_prismer_base',
-        # 'pretrain_prismer_large',
+        'pretrain_prismer_large',
     ]
     for model_name in model_names:
         if pathlib.Path(f'prismer/logging/{model_name}').exists():