Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ import copy
|
|
8 |
keep_layer_count=6
|
9 |
|
10 |
class ByT5ForTextGeotaggingConfig(PretrainedConfig):
|
11 |
-
model_type = "
|
12 |
|
13 |
def __init__(self, n_clusters, model_name_or_path, class_to_location=None, **kwargs):
|
14 |
super(ByT5ForTextGeotaggingConfig, self).__init__(**kwargs)
|
@@ -75,7 +75,7 @@ def geolocate_text_byt5(text):
|
|
75 |
return lat, lon
|
76 |
|
77 |
|
78 |
-
model = ByT5ForTextGeotagging.from_pretrained("byt5-geotagging-
|
79 |
|
80 |
#text = "¡Barcelona es increíble! #VacacionesEnEspaña"
|
81 |
|
|
|
8 |
keep_layer_count=6
|
9 |
|
10 |
class ByT5ForTextGeotaggingConfig(PretrainedConfig):
|
11 |
+
model_type = "byt5_for_text_geotagging"
|
12 |
|
13 |
def __init__(self, n_clusters, model_name_or_path, class_to_location=None, **kwargs):
|
14 |
super(ByT5ForTextGeotaggingConfig, self).__init__(**kwargs)
|
|
|
75 |
return lat, lon
|
76 |
|
77 |
|
78 |
+
model = ByT5ForTextGeotagging.from_pretrained("yachay/byt5-geotagging-es")
|
79 |
|
80 |
#text = "¡Barcelona es increíble! #VacacionesEnEspaña"
|
81 |
|