Update app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 from PIL import Image
 import torch
-from diffusers import DiffusionPipeline
+from transformers import pipeline
 import numpy as np
 import io
 
@@ -18,7 +18,8 @@ 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"
-    pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16).to("cuda")
+    # تأكد من استخدام الكلاس الصحيح من مكتبة transformers إذا كان متاحًا
+    pipe = pipeline("image-to-video", model=model_id, torch_dtype=torch.float16).to("cuda")
     return pipe
 
 pipe = load_model()
@@ -30,26 +31,22 @@ 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).convert("RGB")
+    image = Image.open(uploaded_image)
     st.image(image, caption='Uploaded Image', use_column_width=True)
 
     if st.button('Generate Video'):
         # تحويل الصورة إلى فيديو
         with st.spinner("Generating video..."):
-            # تحويل الصورة إلى 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
-
+            video_frames = pipe(image, num_frames=frame_count)
+        
         st.success("Video generated successfully!")
 
         # عرض الفيديو
-        video_bytes = io.BytesIO()
-        video_frames[0].save(video_bytes, format="mp4")
-        st.video(video_bytes, format="video/mp4")
+        st.video(video_frames[0], format="video/mp4")
 
         # تنزيل الفيديو
+        video_bytes = io.BytesIO()
+        video_frames[0].save(video_bytes, format="mp4")
         st.download_button(label="Download Video", data=video_bytes.getvalue(), file_name="generated_video.mp4", mime="video/mp4")
 
 # تقديم بعض المعلومات حول النموذج