Update app.py

app.py

@@ -87,8 +87,11 @@ def geolocate_text_byt5(text):
     logits, (lat, lon) = model(input_tensor.unsqueeze(0), return_coordinates=True)
     return lat, lon
 
-if 'text_input_state' not in st.session_state:
-    st.session_state.text_input_state = ""
+if 'text_input' not in st.session_state:
+    st.session_state.text_input = ""
+
+if 'text_modified' not in st.session_state:
+    st.session_state.text_modified = ""
 
 # When an example button is clicked, update the session state
 def set_example_text(example_text):
@@ -109,15 +112,14 @@ st.title('GeoTagging using ByT5')
 st.write('Examples:')
 for example in example_texts:
     if st.button(f"{example}"):
-        st.session_state.text_input_state = example
+        st.session_state.text_input = example
+        st.session_state.text_modified = st.session_state.text_input
 
 # 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_state, key='my_text_input')
-if text_input != st.session_state.text_input_state:
-    st.session_state.text_input_state = text_input
+st.session_state.text_modified = st.text_input('Enter your text:', value=st.session_state.text_input)
     
-if text_input:
-    location = geolocate_text_byt5(text_input)
+if st.session_state.text_input:
+    location = geolocate_text_byt5(st.session_state.text_input)
     st.write('Predicted Location: ', location)
 
      # Render map with pydeck