Update app.py

app.py

@@ -16,7 +16,10 @@ if st.button("Generate Video"):
         st.warning("Please upgrade 'accelerate' to version 0.17.0 or higher for CPU offloading.")
     else:
         with st.spinner("Generating video..."):
-            pipe = DiffusionPipeline.from_pretrained("damo-vilab/text-to-video-ms-1.7b", torch_dtype=torch.float16, variant="fp16")
+            pipe = DiffusionPipeline.from_pretrained("damo-vilab/text-to-video-ms-1.7b", 
+                                         torch_dtype=torch.float16, 
+                                         variant="fp16",
+                                         device="cpu") # Force CPU usage
             pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
             pipe.enable_model_cpu_offload()  # Assuming 'accelerate' is updated