Hugging Face's logo

Spaces:
flax-community
/
multilingual-image-captioning
Runtime error

App Files Community
bhavitvyamalik commited on
Commit
2ecf349
·
1 Parent(s): 05cc2da

remove forced download

Browse files
Files changed (1) hide show
  1. apps/mic.py +1 -1
apps/mic.py CHANGED
@@ -44,7 +44,7 @@ def app(state):
44
 
45
  @st.cache
46
  def load_model(ckpt):
47
- return FlaxCLIPVisionMBartForConditionalGeneration.from_pretrained(ckpt, cache_dir="./", force_download=True)
48
 
49
  @st.cache
50
  def generate_sequence(pixel_values, lang_code, num_beams, temperature, top_p, do_sample, top_k, max_length):
 
44
 
45
  @st.cache
46
  def load_model(ckpt):
47
+ return FlaxCLIPVisionMBartForConditionalGeneration.from_pretrained(ckpt)
48
 
49
  @st.cache
50
  def generate_sequence(pixel_values, lang_code, num_beams, temperature, top_p, do_sample, top_k, max_length):