Update pages/01_🦷 Segment.py

pages/01_🦷 Segment.py

@@ -16,7 +16,8 @@ from stpyvista import stpyvista
 
 import torch
 import torch.nn as nn
-import torch.nn.functional as F
+import torch.nn.functional as
+ F
 from torch.autograd import Variable
 
 import streamlit as st
@@ -858,7 +859,7 @@ def segmentation_main(obj_path):
                 # Show the plot
                 #plotter.show()
                 ## Send to streamlit
-                with st.expander("**View Segmentation Result** - ", expanded=False):
+                with st.expander("**View Segmentation Result** - scroll for zoom", expanded=False):
                     stpyvista(plotter)
 
 # Configure Streamlit page
@@ -890,7 +891,7 @@ class Segment(TeethApp):
                 "✔️ Submit",
                 help="Submit 3D scan for segmentation",
             )
-            with st.expander("View Scan", expanded=False):
+            with st.expander("View Scan - scroll for zoom", expanded=False):
                 stpyvista(plotter)
 
             if segment:
@@ -917,7 +918,7 @@ class Segment(TeethApp):
                 "✔️ Submit",
                 help="Submit 3D scan for segmentation",
             )
-                with st.expander("View Scan", expanded=False):
+                with st.expander("View Scan - scroll for zoom", expanded=False):
                     stpyvista(plotter)
 
                 if segment: