Update src/streamlit_app.py

src/streamlit_app.py

@@ -16,15 +16,15 @@ In the meantime, below is an example of what you can do with just a few lines of
 def init_robot_system():
     ae, config = load_saved_model(device="cpu")
     robot_chars = RobotCharacteristics()
-    label_data = np.load('./data/label.npy')
-    njnt_data = np.load('.data/njnt.npy')
-    indices_data = np.load('.data/indices.npy')
+    # label_data = np.load('./src/label.npy')
+    njnt_data = np.load('./src/njnt.npy')
+    # indices_data = np.load('./src/indices.npy')
     robot_chars.max_joints_per_group=njnt_data
     return ae, robot_chars
 motion_level = 'easy'
 x_1 = np.load(f'./src/{motion_level}_X1.npy')
-x_2 = np.load(f'./{motion_level}_X2.npy')
-z_1 = np.load(f'./{motion_level}_Z1.npy')
+x_2 = np.load(f'./src/{motion_level}_X2.npy')
+z_1 = np.load(f'./src/{motion_level}_Z1.npy')
 
 
 
@@ -32,11 +32,11 @@ st.title("LEGO Latent Space Explorer")
 z1 = st.slider("X", -10.0, 10.0)
 z2 = st.slider("Y", -10.0, 10.0)
 
-if st.button("Generate Design"):
+if st.button("Explore Latent Space"):
     ae, robot_chars = init_robot_system()
     decode = ae.decode(th.tensor(easy_optimal,device=next(ae.parameters()).device,dtype=th.float32)).detach().numpy()
     full_decode=robot_chars.build_full_interp_from_symmetric(interp_data_sym=decode,threshold=0.5)
     data_to_visualize = robot_chars.convert_flat_interp_to_dict(full_decode)
     clip = robot_chars.visualize_data_to_classify(data_to_visualize, save_video=True,vis=True, test = True, xml=True)
-    st.success("디자인 생성 완료!")
+    st.success("Kinematic Structure Decoded!")
     st.image("result.png") # 생성된 이미지 보여주기