map

app.py

@@ -51,8 +51,10 @@ def build_map(agent_sp):
             )
         ).add_to(m)
 
-    return m.get_root().render()
-
+    # return m.get_root().render()
+    m.get_root().width = "100%"
+    m.get_root().height = "500px"
+    return m._repr_html_()
 
 def build_poi_sequence(agent_sp):
     lines = []
@@ -90,7 +92,7 @@ with gr.Blocks(title="HiCoTraj Demo", theme=gr.themes.Soft()) as app:
                                  label="Select Agent", value=str(sample_agents[0]))
         demo_label = gr.Textbox(label="Ground Truth Demographics", interactive=False)
 
-    map_out = gr.HTML(label="Trajectory Map")
+    map_out = gr.HTML(label="Trajectory Map", value="<div style='height:500px'></div>")
     poi_out = gr.Textbox(label="POI Sequence", lines=20, interactive=False)
 
     agent_dd.change(fn=on_select, inputs=agent_dd, outputs=[map_out, poi_out, demo_label])