Update app.py

app.py

@@ -26,6 +26,7 @@ try:
     st.write("Cache migration completed successfully.")
 except Exception as e:
     st.error(f"Cache migration failed: {e}")
+    st.write("Proceeding without cache migration...")
 
 if uploaded_file and prompt:
     try:
@@ -41,15 +42,21 @@ if uploaded_file and prompt:
 
         # Load the image
         st.write("Loading image...")
-        image = load_image("uploaded_image.jpg")
-        st.write("Image loaded successfully.")
+        try:
+            image = load_image("uploaded_image.jpg")
+            st.write("Image loaded successfully.")
+        except Exception as e:
+            st.error("Error loading the image.")
+            st.write(f"Debug info: {e}")
+            raise e
 
         # Initialize the CogVideoX pipeline
         st.write("Initializing the pipeline...")
         try:
             pipe = CogVideoXImageToVideoPipeline.from_pretrained(
                 "THUDM/CogVideoX1.5-5B-I2V",
-                torch_dtype=torch.bfloat16
+                torch_dtype=torch.bfloat16,
+                cache_dir="./huggingface_cache"  # Adjust path if necessary
             )
             st.write("Pipeline initialized successfully.")
         except Exception as e:
@@ -67,22 +74,32 @@ if uploaded_file and prompt:
 
         # Generate the video
         st.write("Generating video... This may take a while.")
-        video_frames = pipe(
-            prompt=prompt,
-            image=image,
-            num_videos_per_prompt=1,
-            num_inference_steps=50,
-            num_frames=81,
-            guidance_scale=6,
-            generator=torch.Generator(device="cuda").manual_seed(42),
-        ).frames[0]
-        st.write("Video generated successfully.")
+        try:
+            video_frames = pipe(
+                prompt=prompt,
+                image=image,
+                num_videos_per_prompt=1,
+                num_inference_steps=50,
+                num_frames=81,
+                guidance_scale=6,
+                generator=torch.Generator(device="cuda").manual_seed(42),
+            ).frames[0]
+            st.write("Video generated successfully.")
+        except Exception as e:
+            st.error("Error during video generation.")
+            st.write(f"Debug info: {e}")
+            raise e
 
         # Export video
         st.write("Exporting video...")
-        video_path = "output.mp4"
-        export_to_video(video_frames, video_path, fps=8)
-        st.write("Video exported successfully.")
+        try:
+            video_path = "output.mp4"
+            export_to_video(video_frames, video_path, fps=8)
+            st.write("Video exported successfully.")
+        except Exception as e:
+            st.error("Error exporting video.")
+            st.write(f"Debug info: {e}")
+            raise e
 
         # Display video in Streamlit
         st.video(video_path)