yachay commited on
Commit
ff38b8f
·
1 Parent(s): fe60167

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -74,9 +74,8 @@ def geolocate_text_byt5(text):
74
  logits, (lat, lon) = model(input_tensor.unsqueeze(0), return_coordinates=True)
75
  return lat, lon
76
 
77
-
78
- model = ByT5ForTextGeotagging.from_pretrained("yachay/byt5-geotagging-es", token="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")
79
  byt5_tokenizer = AutoTokenizer.from_pretrained("yachay/byt5-geotagging-es", token="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")
 
80
 
81
  #text = "¡Barcelona es increíble! #VacacionesEnEspaña"
82
 
 
74
  logits, (lat, lon) = model(input_tensor.unsqueeze(0), return_coordinates=True)
75
  return lat, lon
76
 
 
 
77
  byt5_tokenizer = AutoTokenizer.from_pretrained("yachay/byt5-geotagging-es", token="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")
78
+ model = ByT5ForTextGeotagging.from_pretrained("yachay/byt5-geotagging-es", token="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")
79
 
80
  #text = "¡Barcelona es increíble! #VacacionesEnEspaña"
81