Update app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 from PIL import Image
 import torch
-from transformers import pipeline
+from diffusers import DiffusionPipeline
 import numpy as np
 import io
 
@@ -18,8 +18,7 @@ st.write("Upload an image to generate a video. You can also adjust settings for
 @st.cache_resource
 def load_model():
     model_id = "stabilityai/stable-video-diffusion-img2vid-xt"
-    # تأكد من استخدام الكلاس الصحيح من مكتبة transformers إذا كان متاحًا
-    pipe = pipeline("image-to-video", model=model_id, torch_dtype=torch.float16).to("cuda")
+    pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16).to("cuda")
     return pipe
 
 pipe = load_model()
@@ -31,22 +30,26 @@ uploaded_image = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "pn
 frame_count = st.slider("Number of frames", min_value=10, max_value=50, value=25, step=5)
 
 if uploaded_image is not None:
-    image = Image.open(uploaded_image)
+    image = Image.open(uploaded_image).convert("RGB")
     st.image(image, caption='Uploaded Image', use_column_width=True)
 
     if st.button('Generate Video'):
         # تحويل الصورة إلى فيديو
         with st.spinner("Generating video..."):
-            video_frames = pipe(image, num_frames=frame_count)
-        
+            # تحويل الصورة إلى tensor
+            image_tensor = torch.tensor(np.array(image)).float().unsqueeze(0).permute(0, 3, 1, 2).to("cuda") / 255.0
+            
+            # توليد الفيديو
+            video_frames = pipe(image_tensor, num_frames=frame_count).images
+
         st.success("Video generated successfully!")
 
         # عرض الفيديو
-        st.video(video_frames[0], format="video/mp4")
-
-        # تنزيل الفيديو
         video_bytes = io.BytesIO()
         video_frames[0].save(video_bytes, format="mp4")
+        st.video(video_bytes, format="video/mp4")
+
+        # تنزيل الفيديو
         st.download_button(label="Download Video", data=video_bytes.getvalue(), file_name="generated_video.mp4", mime="video/mp4")
 
 # تقديم بعض المعلومات حول النموذج