yachay commited on
Commit
160efbe
·
1 Parent(s): b99d24c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -73,7 +73,7 @@ class ByT5ForTextGeotagging(PreTrainedModel):
73
  return logits
74
 
75
 
76
- @st.cache(allow_output_mutation=True)
77
  def load_model_and_tokenizer():
78
  byt5_tokenizer = AutoTokenizer.from_pretrained("yachay/byt5-geotagging-es", token="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")
79
  model = ByT5ForTextGeotagging.from_pretrained("yachay/byt5-geotagging-es", token="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")
 
73
  return logits
74
 
75
 
76
+ @st.cache_data(ttl=None, persist=True)
77
  def load_model_and_tokenizer():
78
  byt5_tokenizer = AutoTokenizer.from_pretrained("yachay/byt5-geotagging-es", token="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")
79
  model = ByT5ForTextGeotagging.from_pretrained("yachay/byt5-geotagging-es", token="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")