Spaces:
Runtime error
Runtime error
File size: 2,549 Bytes
1e350c6 3179455 1e350c6 6641dc5 1e350c6 3179455 e986028 6641dc5 e986028 3179455 1e350c6 f9a94e1 673c700 6641dc5 3179455 e8b2b98 e986028 6641dc5 e986028 1e350c6 e986028 6641dc5 dc76dee 6641dc5 e986028 1e350c6 e986028 62ad765 62cd675 3179455 62cd675 3179455 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 |
import os
import uuid
import streamlit as st
import torch
from diffusers.utils import load_image
# Versuch, die Pipeline zu importieren
try:
from diffusers import CogVideoXImageToVideoPipeline
pipeline_available = True
except ImportError:
pipeline_available = False
st.error(
"Failed to import `CogVideoXImageToVideoPipeline`. "
"Please run `pip install diffusers`."
)
st.title("Image to Video with Hugging Face")
st.write("Upload an image and provide a prompt to generate a video.")
if pipeline_available:
uploaded_file = st.file_uploader(
"Upload an image (JPG or PNG):", type=["jpg", "jpeg", "png"]
)
prompt = st.text_input(
"Enter your prompt:",
"A little girl is riding a bicycle at high speed. Focused, detailed, realistic."
)
if uploaded_file and prompt:
try:
# Save uploaded file
file_name = f"{uuid.uuid4()}_uploaded_image.jpg"
with open(file_name, "wb") as f:
f.write(uploaded_file.read())
st.write("Uploaded image saved successfully.")
# Load the image
image = load_image(file_name)
# Initialize pipeline
device = "cuda" if torch.cuda.is_available() else "cpu"
pipe = CogVideoXImageToVideoPipeline.from_pretrained(
"THUDM/CogVideoX1.5-5B-I2V",
torch_dtype=torch.bfloat16,
cache_dir="./huggingface_cache",
)
pipe.enable_sequential_cpu_offload()
pipe.vae.enable_tiling()
pipe.vae.enable_slicing()
# Generate video
with st.spinner("Generating video... This may take a while."):
try:
# Attempt to generate the video
video_frames = pipe(
prompt=prompt,
image=image,
num_videos_per_prompt=1,
num_inference_steps=50,
num_frames=81,
guidance_scale=6,
generator=torch.Generator(device=device).manual_seed(42),
).frames[0]
st.video(video_frames)
st.success("Video generated successfully!")
except Exception as e:
st.error(f"An error occurred during video generation: {e}")
except Exception as e:
st.error(f"An error occurred while processing the uploaded file: {e}") |