Update app.py

app.py

@@ -326,13 +326,15 @@ with gr.Blocks(title="FoldMark", css=custom_css) as demo:
             #structure_view = gr.HTML(label="3D Visualization")
 
         with gr.Row():
-            view3d = Molecule3D(label="3D Visualization", reps=reps)
+            view3d = Molecule3D(label="3D Visualization(Gray: Unwatermarked; Cyan: Watermarked)", reps=reps)
+            
         legend = gr.Markdown("""
         **Color Legend:**
 
         - <span style="color:grey">Gray: Unwatermarked Structure</span>
         - <span style="color:cyan">Cyan: Watermarked Structure</span>
         """)
+        
         with gr.Row():
             cif_file = gr.File(label="Download CIF File")
         with gr.Row():