Update app.py

app.py

@@ -1,4 +1,6 @@
-import streamlit as st
+from diffusers import DiffusionPipeline
 
-x = st.slider('Select a value')
-st.write(x, 'squared is', x * x)
+pipe = DiffusionPipeline.from_pretrained("Wan-AI/Wan2.1-I2V-14B-720P")
+
+prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
+image = pipe(prompt).images[0]