Update app.py

app.py

@@ -1,15 +1,27 @@
 import streamlit as st
-
-# x = st.slider('Select a value')
-# st.write(x, 'squared is', x * x)
 import torch
 from diffusers import DiffusionPipeline, DPMSolverMultistepScheduler
 from diffusers.utils import export_to_video
 
-pipe = DiffusionPipeline.from_pretrained("damo-vilab/text-to-video-ms-1.7b", torch_dtype=torch.float16, variant="fp16")
-pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
-pipe.enable_model_cpu_offload()
+# Explicitly set the device to CPU
+device = torch.device("cpu")
+
+# Load the model onto the CPU
+pipe = DiffusionPipeline.from_pretrained(
+    "damo-vilab/text-to-video-ms-1.7b", 
+    torch_dtype=torch.float32  # Use float32 for CPU
+).to(device)
+
+# Set the scheduler to use CPU
+pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config).to(device)
+
+# Disable model offloading if running on CPU
+# pipe.enable_model_cpu_offload()  # This line should be removed or commented out
 
 prompt = "Pop international experimental music"
-video_frames = pipe(prompt, num_inference_steps=25).frames
+
+# Generate the video frames on the CPU
+video_frames = pipe(prompt, num_inference_steps=25, device=device).frames
+
+# Export the frames to a video file
 video_path = export_to_video(video_frames)