yachay commited on
Commit
4f08151
·
1 Parent(s): 8358faa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,7 +8,7 @@ import copy
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,7 +75,7 @@ def geolocate_text_byt5(text):
75
  return lat, lon
76
 
77
 
78
- model = ByT5ForTextGeotagging.from_pretrained("byt5-geotagging-spanish")
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