Update app.py

app.py

@@ -4,6 +4,7 @@ from transformers import PretrainedConfig, PreTrainedModel, T5EncoderModel, Auto
 import torch
 import torch.nn as nn
 import copy
+import pydeck as pdk
 
 keep_layer_count=6
 byt5_tokenizer = AutoTokenizer.from_pretrained("yachay/byt5-geotagging-es", token="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")
@@ -85,4 +86,28 @@ st.title('GeoTagging using ByT5')
 text_input = st.text_input('Enter your text:')
 if text_input:
     location = geolocate_text_byt5(text_input)
-    st.write('Predicted Location: ', location)
+    st.write('Predicted Location: ', location)
+     # Render map with pydeck
+    map_data = pd.DataFrame(
+        [[location[0], location[1]]],
+        columns=["lat", "lon"]
+    )
+
+    st.pydeck_chart(pdk.Deck(
+        map_style='mapbox://styles/mapbox/light-v9',
+        initial_view_state=pdk.ViewState(
+            latitude=location[0],
+            longitude=location[1],
+            zoom=11,
+            pitch=50,
+        ),
+        layers=[
+            pdk.Layer(
+                'ScatterplotLayer',
+                data=map_data,
+                get_position='[lon, lat]',
+                get_color='[200, 30, 0, 160]',
+                get_radius=200,
+            ),
+        ],
+    ))