Update app.py

app.py

@@ -1,5 +1,8 @@
 import streamlit as st
 import replicate
+import requests
+from PIL import Image
+from io import BytesIO
 
 # Set the replicate API token
 import os
@@ -18,7 +21,7 @@ st.write("Fill in the input parameters and click on the button to run the Stable
 cond_aug = st.number_input("Conditioning Augmentation", value=0.02)
 decoding_t = st.number_input("Decoding Time", value=7)
 input_image = st.text_input("Input Image URL", "https://example.com/s-widua-icjKmXvsO7U-unsplash.jpg")
-video_length = st.selectbox("Video Length", ["14_frames_with_svd", "other_options"])
+video_length = st.selectbox("Video Length", ["14_frames_with_svd", "25_frames_with_svd_xt"])
 sizing_strategy = st.selectbox("Sizing Strategy", ["maintain_aspect_ratio", "other_options"])
 motion_bucket_id = st.number_input("Motion Bucket ID", value=127)
 frames_per_second = st.number_input("Frames Per Second", value=6)
@@ -35,5 +38,11 @@ input_params = {
 
 if st.button("Run Model"):
     output_result = run_replicate_model(input_params)
-    st.write("Output:")
-    st.code(output_result, language="json")
+    st.write("Output Video:")
+    
+    try:
+        response = requests.get(output_result)
+        video_bytes = BytesIO(response.content)
+        st.video(video_bytes)
+    except Exception as e:
+        st.write("Unable to display the video.")