debug: space requirements

app.py

@@ -1,10 +1,27 @@
+import sys
+import subprocess
+import importlib.util
+
+# Check if required packages are installed
+required_packages = ["ftfy", "einops", "imageio", "imageio-ffmpeg"]
+for package in required_packages:
+    if importlib.util.find_spec(package) is None:
+        print(f"Installing missing dependency: {package}")
+        subprocess.check_call([sys.executable, "-m", "pip", "install", package])
+
 import torch
 import gradio as gr
 import spaces
 from diffusers.utils import export_to_video
-from diffusers import AutoencoderKLWan, WanPipeline
-from diffusers.schedulers.scheduling_unipc_multistep import UniPCMultistepScheduler
-from diffusers.schedulers.scheduling_flow_match_euler_discrete import FlowMatchEulerDiscreteScheduler
+
+# Now try to import the specific components
+try:
+    from diffusers import AutoencoderKLWan, WanPipeline
+    from diffusers.schedulers.scheduling_unipc_multistep import UniPCMultistepScheduler
+    from diffusers.schedulers.scheduling_flow_match_euler_discrete import FlowMatchEulerDiscreteScheduler
+except ImportError as e:
+    print(f"Error importing diffusers components: {e}")
+    subprocess.check_call([sys.executable, "-m", "pip", "install", "--upgrade", "diffusers"])
 
 # Define model options
 MODEL_OPTIONS = {

requirements.txt

@@ -1,10 +1,11 @@
-git+https://github.com/huggingface/diffusers.git
-transformers
-accelerate
-safetensors
+diffusers==0.27.1
+transformers>=4.36.0
+accelerate>=0.27.2
+safetensors>=0.4.1
 torch>=2.0.1
-ftfy
-einops
-imageio
-imageio-ffmpeg
-opencv-python
+ftfy>=6.1.3
+einops>=0.7.0
+imageio>=2.31.6
+imageio-ffmpeg>=0.4.9
+opencv-python>=4.9.0.0
+omegaconf>=2.3.0