Upload 3 files

app.py

@@ -17,6 +17,31 @@ packages_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'package
 if os.path.exists(packages_dir):
     sys.path.append(packages_dir)
 
+# Check if complex dependencies are installed
+def check_complex_dependencies():
+    """Check if complex dependencies are available"""
+    try:
+        import torch_sparse
+        import torch_scatter
+        import nvdiffrast
+        import pytorch3d
+        return True
+    except ImportError:
+        return False
+
+# Run post-install if needed
+if not check_complex_dependencies():
+    print("Complex dependencies not found, running post-install...")
+    try:
+        import subprocess
+        subprocess.run([sys.executable, "post_install.py"], check=True)
+        print("Post-install completed, restarting...")
+        # Restart the process to pick up new dependencies
+        os.execv(sys.executable, ['python'] + sys.argv)
+    except Exception as e:
+        print(f"Post-install failed: {e}")
+        sys.exit(1)
+
 try:
     from loop import loop
 except ImportError as e:

post_install.py

@@ -0,0 +1,157 @@
+#!/usr/bin/env python3
+"""
+Post-install script for Hugging Face Spaces
+This script installs complex dependencies that need PyTorch to be available first
+"""
+
+import os
+import sys
+import subprocess
+import shutil
+from pathlib import Path
+
+def run_command(command, cwd=None):
+    """Run a shell command and return the result"""
+    print(f"Running: {command}")
+    result = subprocess.run(command, shell=True, cwd=cwd, capture_output=True, text=True)
+    if result.returncode != 0:
+        print(f"Error running command: {command}")
+        print(f"Error output: {result.stderr}")
+        return False
+    print(f"Success: {command}")
+    return True
+
+def install_torch_sparse():
+    """Install torch-sparse with the current PyTorch version"""
+    print("Installing torch-sparse...")
+    
+    # Get current PyTorch version
+    try:
+        import torch
+        torch_version = torch.__version__
+        print(f"Current PyTorch version: {torch_version}")
+        
+        # Extract major.minor version
+        version_parts = torch_version.split('.')
+        torch_major_minor = f"{version_parts[0]}.{version_parts[1]}"
+        
+        # Try to install torch-sparse with the correct version
+        if not run_command(f"pip install torch-sparse -f https://data.pyg.org/whl/torch-{torch_major_minor}+cu121.html"):
+            print("Failed to install torch-sparse with pre-built wheel, trying from source...")
+            if not run_command("pip install torch-sparse"):
+                return False
+        
+        return True
+    except Exception as e:
+        print(f"Error installing torch-sparse: {e}")
+        return False
+
+def install_torch_scatter():
+    """Install torch-scatter with the current PyTorch version"""
+    print("Installing torch-scatter...")
+    
+    try:
+        import torch
+        torch_version = torch.__version__
+        version_parts = torch_version.split('.')
+        torch_major_minor = f"{version_parts[0]}.{version_parts[1]}"
+        
+        if not run_command(f"pip install torch-scatter -f https://data.pyg.org/whl/torch-{torch_major_minor}+cu121.html"):
+            print("Failed to install torch-scatter with pre-built wheel, trying from source...")
+            if not run_command("pip install torch-scatter"):
+                return False
+        
+        return True
+    except Exception as e:
+        print(f"Error installing torch-scatter: {e}")
+        return False
+
+def install_nvdiffrast():
+    """Install nvdiffrast"""
+    print("Installing nvdiffrast...")
+    
+    # Create packages directory if it doesn't exist
+    packages_dir = Path("packages")
+    packages_dir.mkdir(exist_ok=True)
+    
+    # Clone nvdiffrast
+    if not (packages_dir / "nvdiffrast").exists():
+        if not run_command("git clone https://github.com/NVlabs/nvdiffrast.git", cwd=packages_dir):
+            return False
+    
+    # Install nvdiffrast
+    nvdiffrast_dir = packages_dir / "nvdiffrast"
+    if not run_command("pip install .", cwd=nvdiffrast_dir):
+        return False
+    
+    return True
+
+def install_pytorch3d():
+    """Install PyTorch3D"""
+    print("Installing PyTorch3D...")
+    
+    packages_dir = Path("packages")
+    
+    # Clone PyTorch3D
+    if not (packages_dir / "pytorch3d").exists():
+        if not run_command("git clone https://github.com/facebookresearch/pytorch3d.git", cwd=packages_dir):
+            return False
+    
+    # Install PyTorch3D with CUDA support
+    pytorch3d_dir = packages_dir / "pytorch3d"
+    if not run_command("FORCE_CUDA=1 pip install .", cwd=pytorch3d_dir):
+        return False
+    
+    return True
+
+def install_fashion_clip():
+    """Setup Fashion-CLIP"""
+    print("Setting up Fashion-CLIP...")
+    
+    packages_dir = Path("packages")
+    
+    # Clone Fashion-CLIP if not already present
+    if not (packages_dir / "fashion-clip").exists():
+        if not run_command("git clone https://github.com/patrickjohncyh/fashion-clip.git", cwd=packages_dir):
+            return False
+    
+    # Install Fashion-CLIP dependencies
+    fashion_clip_dir = packages_dir / "fashion-clip"
+    dependencies = ["appdirs", "boto3", "annoy", "validators", "transformers", "datasets"]
+    
+    for dep in dependencies:
+        if not run_command(f"pip install {dep}", cwd=fashion_clip_dir):
+            print(f"Warning: Failed to install {dep}")
+    
+    return True
+
+def main():
+    """Main installation function"""
+    print("Starting post-installation for Garment3DGen...")
+    
+    # Install complex dependencies
+    if not install_torch_sparse():
+        print("Failed to install torch-sparse")
+        sys.exit(1)
+    
+    if not install_torch_scatter():
+        print("Failed to install torch-scatter")
+        sys.exit(1)
+    
+    if not install_nvdiffrast():
+        print("Failed to install nvdiffrast")
+        sys.exit(1)
+    
+    if not install_pytorch3d():
+        print("Failed to install PyTorch3D")
+        sys.exit(1)
+    
+    if not install_fashion_clip():
+        print("Failed to install Fashion-CLIP")
+        sys.exit(1)
+    
+    print("Post-installation completed successfully!")
+    print("All dependencies are now available.")
+
+if __name__ == "__main__":
+    main() 

requirements.txt

@@ -1,4 +1,5 @@
-# Core dependencies (PyTorch will be installed separately in Dockerfile)
+# Core dependencies - compatible with HF Spaces PyTorch 2.7.1
+# PyTorch will be installed by HF Spaces automatically
 clip
 imageio
 cython
@@ -33,5 +34,5 @@ boto3
 annoy
 validators
 
-# Note: PyTorch, torch-sparse, torch-scatter, pytorch3d, and nvdiffrast 
+# Note: torch-sparse, torch-scatter, pytorch3d, and nvdiffrast 
 # will be installed separately in the Dockerfile to ensure correct versions