Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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()
|