evelynsaltt commited on
Commit
706f673
·
verified ·
1 Parent(s): fd1defa

update demo code

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +21 -25
src/streamlit_app.py CHANGED
@@ -13,28 +13,24 @@ forums](https://discuss.streamlit.io).
13
  In the meantime, below is an example of what you can do with just a few lines of code:
14
  """
15
 
16
- num_points = st.slider("Number of points in spiral", 1, 10000, 1100)
17
- num_turns = st.slider("Number of turns in spiral", 1, 300, 31)
18
-
19
- indices = np.linspace(0, 1, num_points)
20
- theta = 2 * np.pi * num_turns * indices
21
- radius = indices
22
-
23
- x = radius * np.cos(theta)
24
- y = radius * np.sin(theta)
25
-
26
- df = pd.DataFrame({
27
- "x": x,
28
- "y": y,
29
- "idx": indices,
30
- "rand": np.random.randn(num_points),
31
- })
32
-
33
- st.altair_chart(alt.Chart(df, height=700, width=700)
34
- .mark_point(filled=True)
35
- .encode(
36
- x=alt.X("x", axis=None),
37
- y=alt.Y("y", axis=None),
38
- color=alt.Color("idx", legend=None, scale=alt.Scale()),
39
- size=alt.Size("rand", legend=None, scale=alt.Scale(range=[1, 150])),
40
- ))
 
13
  In the meantime, below is an example of what you can do with just a few lines of code:
14
  """
15
 
16
+
17
+ # Set your video directory here
18
+ VIDEO_FOLDER = "synthda_fallingdemo_realreal"
19
+
20
+ st.set_page_config(layout="centered")
21
+ st.title("AutoSynthDa Pose Interpolation Viewer")
22
+
23
+ st.markdown("Use the slider to explore how pose interpolation changes as the weight increases.")
24
+
25
+ # Slider for selecting weight
26
+ weight = st.slider("Interpolation Weight", 0.0, 1.0, step=0.1)
27
+
28
+ # Construct filename based on weight (rounded to one decimal for file matching)
29
+ filename = f"videos_generated_{weight:.1f}.mp4"
30
+ video_path = os.path.join(VIDEO_FOLDER, filename)
31
+
32
+ # Check if the video exists
33
+ if os.path.exists(video_path):
34
+ st.video(video_path)
35
+ else:
36
+ st.error(f"No video found for weight = {weight:.1f}")