Update app.py

app.py

@@ -44,12 +44,15 @@ def sample_info(sample_id_str, fieldn):
     triangles = plaid_sample.get_elements()['TRI_3']
 
     # generate colormap
-    norm = mpl.colors.Normalize(vmin=np.min(field), vmax=np.max(field))
-    cmap = cm.coolwarm
-
-    m = cm.ScalarMappable(norm=norm, cmap=cmap)
-    vertex_colors = m.to_rgba(field)[:,:3]
-
+    if np.linalg.norm(field) > 0:
+        norm = mpl.colors.Normalize(vmin=np.min(field), vmax=np.max(field))
+        cmap = cm.coolwarm
+        m = cm.ScalarMappable(norm=norm, cmap=cmap)
+    
+        vertex_colors = m.to_rgba(field)[:,:3]
+    else:
+        vertex_colors = np.zeros((field.shape[0], 3))
+    
     # generate mesh
     trimesh = Trimesh(vertices = nodes, faces = triangles)
     trimesh.visual.vertex_colors = vertex_colors
@@ -82,7 +85,7 @@ def sample_info(sample_id_str, fieldn):
 if __name__ == "__main__":
 
     with gr.Blocks() as demo:
-        d1 = gr.Slider(0, 499, value=0, label="Training sample id", info="Choose between 0 and 999")
+        d1 = gr.Slider(0, 499, value=0, label="Training sample id", info="Choose between 0 and 499")
         d2 = gr.Dropdown(field_names_train, value=field_names_train[0], label="Field name")