synthda-demo / src /streamlit_app.py
evelynsaltt's picture
Update src/streamlit_app.py
641ce46 verified
raw
history blame
1.3 kB
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("""
# AutoSynthDa Interpolation Viewer
AutoSynthDa blends two input motion videos to **generate kinematically coherent, synthetic action videos**.
Move the slider to explore how motion changes from one input to another.
View code and try the code on Colab at github.com/nvidia/synthda
""")
# Slider for selecting weight
weight = st.slider("Interpolation Weight", 0.0, 1.0, step=0.1)
# Label logic
if weight == 0.0:
st.success("Showing: Input Video 1 (No interpolation)")
elif weight == 1.0:
st.success("Showing: Input Video 2 (No interpolation)")
else:
w2 = round(1.0 - weight, 1)
st.info(f"Generating motion by blending {weight:.1f} of Input Video 1 and {w2:.1f} of Input Video 2")
# 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)
# Show video
if os.path.exists(video_path):
st.video(video_path)
else:
st.error(f"No video found for weight = {weight:.1f}")