Update app.py

app.py

@@ -11,9 +11,9 @@ st.title("🚀 Text-to-Video (Zeroscope)")
 def load_model():
     pipe = DiffusionPipeline.from_pretrained(
         "cerspense/zeroscope_v2_576w", 
-        torch_dtype=torch.float16
+        torch_dtype=torch.float32
     )
-    pipe.enable_cpu_offload()
+    pipe.to("cpu")
     return pipe
 
 pipe = load_model()
@@ -22,16 +22,9 @@ prompt = st.text_area("Enter prompt (short & descriptive):", max_chars=50)
 
 if st.button("Generate Video"):
     if prompt:
-        with st.spinner("Generating... (this may take ~2-3 mins)"):
-            video_frames = pipe(prompt, num_frames=10, height=320, width=576).frames
+        with st.spinner("Generating... (may take a few mins on CPU)"):
+            video_frames = pipe(prompt, num_frames=8, height=320, width=576).frames
 
             video_filename = tempfile.mktemp(".mp4")
             clips = [ImageClip(np.array(frame)).set_duration(0.3) for frame in video_frames]
-            final_clip = concatenate_videoclips(clips, method="compose")
-            final_clip.write_videofile(video_filename, fps=5)
-
-            st.video(video_filename)
-
-            os.remove(video_filename)
-    else:
-        st.warning("Enter a prompt to generate a video.")
+            final_clip = concatenate_videoclips(clips, method="compose")