synthda-demo / src /streamlit_app.py
evelynsaltt's picture
Update src/streamlit_app.py
b65af1e verified
raw
history blame contribute delete
801 Bytes
import altair as alt
import numpy as np
import pandas as pd
import streamlit as st
import os
# Set your video directory here
VIDEO_FOLDER = "./src/synthda_falling_realreal/"
st.set_page_config(layout="centered")
st.title("AutoSynthDa Pose Interpolation Viewer")
st.markdown("Use the slider to explore how pose interpolation changes as the weight increases.")
# Slider for selecting weight
weight = st.slider("Interpolation Weight", 0.0, 1.0, step=0.1)
# Construct filename based on weight (rounded to one decimal for file matching)
filename = f"videos_generated_{weight:.1f}.mp4"
video_path = os.path.join(VIDEO_FOLDER, filename)
print(video_path)
# Check if the video exists
if os.path.exists(video_path):
st.video(video_path)
else:
st.error(f"No video found for weight = {weight:.1f}")