yachay commited on
Commit
6f8ed7d
·
1 Parent(s): 190ce08

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -76,8 +76,8 @@ class ByT5ForTextGeotagging(PreTrainedModel):
76
 
77
  @st.cache_data(ttl=None, persist=True)
78
  def load_model_and_tokenizer():
79
- byt5_tokenizer = AutoTokenizer.from_pretrained("yachay/byt5-geotagging-es", st.secrets['token'])
80
- model = ByT5ForTextGeotagging.from_pretrained("yachay/byt5-geotagging-es", st.secrets['token'])
81
  return byt5_tokenizer, model
82
 
83
  byt5_tokenizer, model = load_model_and_tokenizer()
 
76
 
77
  @st.cache_data(ttl=None, persist=True)
78
  def load_model_and_tokenizer():
79
+ byt5_tokenizer = AutoTokenizer.from_pretrained("yachay/byt5-geotagging-es", token=st.secrets['token'])
80
+ model = ByT5ForTextGeotagging.from_pretrained("yachay/byt5-geotagging-es", token=st.secrets['token'])
81
  return byt5_tokenizer, model
82
 
83
  byt5_tokenizer, model = load_model_and_tokenizer()