force download

apps/mic.py

@@ -44,7 +44,7 @@ def app(state):
 
     @st.cache
     def load_model(ckpt):
-        return FlaxCLIPVisionMBartForConditionalGeneration.from_pretrained(ckpt)
+        return FlaxCLIPVisionMBartForConditionalGeneration.from_pretrained(ckpt, cache_dir="./", force_download=True)
 
     @st.cache
     def generate_sequence(pixel_values, lang_code, num_beams, temperature, top_p, do_sample, top_k, max_length):
@@ -54,7 +54,7 @@ def app(state):
         output_sequence = tokenizer.batch_decode(output_ids[0], skip_special_tokens=True, max_length=max_length)
         return output_sequence
 
-    mic_checkpoints = ["./ckpts/ckpt-51999"]  # TODO: Maybe add more checkpoints?
+    mic_checkpoints = ["flax-community/clip-vit-base-patch32_mbart-large-50"]  # TODO: Maybe add more checkpoints?
     dummy_data = pd.read_csv("reference.tsv", sep="\t")
 
     first_index = 25