Update app.py

app.py

@@ -8,7 +8,6 @@ import copy
 import pydeck as pdk
 
 keep_layer_count=6
-byt5_tokenizer = AutoTokenizer.from_pretrained("yachay/byt5-geotagging-es", token="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")
 
 
 class ByT5ForTextGeotaggingConfig(PretrainedConfig):
@@ -73,13 +72,27 @@ class ByT5ForTextGeotagging(PreTrainedModel):
         else:
             return logits
 
+
+@st.cache(allow_output_mutation=True)
+def load_model_and_tokenizer():
+    byt5_tokenizer = AutoTokenizer.from_pretrained("yachay/byt5-geotagging-es", token="x")
+    model = ByT5ForTextGeotagging.from_pretrained("yachay/byt5-geotagging-es", token="x")
+    return byt5_tokenizer, model
+
+byt5_tokenizer, model = load_model_and_tokenizer()
+
+
 def geolocate_text_byt5(text):
     input_tensor = byt5_tokenizer(text, return_tensors="pt", truncation=True, max_length=140)['input_ids']
     logits, (lat, lon) = model(input_tensor.unsqueeze(0), return_coordinates=True)
     return lat, lon
 
-model = ByT5ForTextGeotagging.from_pretrained("yachay/byt5-geotagging-es", token="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")
+if 'text_input' not in st.session_state:
+    st.session_state.text_input = ""
 
+# When an example button is clicked, update the session state
+def set_example_text(example_text):
+    st.session_state.text_input = example_text
 
 example_texts = [
     "Disfrutando de una paella deliciosa en las playas de #Valencia 🥘☀️",
@@ -93,17 +106,18 @@ example_texts = [
 # Streamlit interface
 st.title('GeoTagging using ByT5')
 
-# Buttons for example texts
-for ex_text in example_texts:
-    if st.button(f'Example: "{ex_text[:30]}..."'):
-        text_input = ex_text
-
-text_input = st.text_input('Enter your text:', value=text_input if 'text_input' in locals() else '')
+for example in example_texts:
+    if st.button(f"Use example: {example}"):
+        set_example_text(example)
 
+# Get text input and update session state when it's modified
+text_input = st.text_input('Enter your text:', value=st.session_state.text_input)
+st.session_state.text_input = text_input
 
 if text_input:
     location = geolocate_text_byt5(text_input)
     st.write('Predicted Location: ', location)
+
      # Render map with pydeck
     map_data = pd.DataFrame(
         [[location[0], location[1]]],