Update app.py

app.py

@@ -9,6 +9,12 @@ from PIL import Image
 import json
 
 # Project by Nymbo
+from diffusers import DiffusionPipeline
+
+pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-video-diffusion-img2vid-xt")
+
+prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
+image = pipe(prompt).images[0]
 
 API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-schnell"
 API_TOKEN = os.getenv("HF_READ_TOKEN")