Replace install_croco function with a direct pip installation command for the curope model dependency in app.py, streamlining the installation process and removing the previous compilation logic.

app.py

@@ -1,129 +1,13 @@
 import subprocess
 import sys
-import spaces
-import torch
 
-@spaces.GPU
-@torch.autocast("cuda")
-@torch.no_grad()
-def install_croco():
-    """
-    Compile curope CUDA extension using torch.utils.cpp_extension.load()
-    This approach works better with zerogpu than subprocess calls.
-    """
-    try:
-        import os
-        import shutil
-        from torch.utils.cpp_extension import load
-        
-        # Path to the curope source files
-        curope_path = "./extern/CUT3R/src/croco/models/curope"
-        
-        # Check if source files exist
-        cpp_file = os.path.join(curope_path, "curope.cpp")
-        cu_file = os.path.join(curope_path, "kernels.cu")
-        
-        if not os.path.exists(cpp_file) or not os.path.exists(cu_file):
-            print(f"Source files not found: {cpp_file}, {cu_file}")
-            return False
-        
-        print(f"Found source files: {cpp_file}, {cu_file}")
-        
-        # Define source files
-        sources = [cpp_file, cu_file]
-        
-        # Get CUDA architectures - be more conservative
-        if torch.cuda.is_available():
-            try:
-                # Get the current GPU architecture
-                capability = torch.cuda.get_device_capability()
-                arch_code = f"compute_{capability[0]}{capability[1]}"
-                sm_code = f"sm_{capability[0]}{capability[1]}"
-                all_cuda_archs = ['-gencode', f'arch={arch_code},code={sm_code}']
-                print(f"Using CUDA architecture: {arch_code}")
-            except:
-                # Fallback to common architectures
-                all_cuda_archs = [
-                    '-gencode', 'arch=compute_70,code=sm_70',
-                    '-gencode', 'arch=compute_80,code=sm_80',
-                ]
-        else:
-            print("CUDA not available!")
-            return False
-        
-        # Create build directory in current working directory
-        build_dir = os.path.abspath("./curope_build_temp")
-        if os.path.exists(build_dir):
-            shutil.rmtree(build_dir)
-        os.makedirs(build_dir, exist_ok=True)
-        print(f"Using build directory: {build_dir}")
-        
-        # Compile the extension
-        print("Starting CUDA compilation...")
-        curope = load(
-            name="curope",
-            sources=sources,
-            extra_cflags=["-O3"],
-            extra_cuda_cflags=["-O3", "--use_fast_math"] + all_cuda_archs,
-            verbose=True,
-            build_directory=build_dir,
-            with_cuda=True
-        )
-        
-        print("Successfully compiled curope CUDA extension")
-        
-        # Clean up build directory
-        try:
-            shutil.rmtree(build_dir)
-        except:
-            pass
-            
-        return True
-        
-    except Exception as e:
-        print(f"Method 1 failed: {e}")
-        print("Trying alternative compilation method...")
-        
-        # Alternative method: use setup.py build_ext --inplace
-        try:
-            import subprocess
-            import sys
-            import os
-            
-            # Change to curope directory
-            original_cwd = os.getcwd()
-            curope_path = "./extern/CUT3R/src/croco/models/curope"
-            os.chdir(curope_path)
-            
-            # Set CUDA environment variables to ensure they're available
-            env = os.environ.copy()
-            if torch.cuda.is_available():
-                cuda_home = os.environ.get('CUDA_HOME') or os.environ.get('CUDA_PATH')
-                if cuda_home:
-                    env['CUDA_HOME'] = cuda_home
-                    env['CUDA_PATH'] = cuda_home
-            
-            # Run setup.py build_ext --inplace within the GPU context
-            result = subprocess.run([
-                sys.executable, "setup.py", "build_ext", "--inplace"
-            ], env=env, capture_output=True, text=True)
-            
-            os.chdir(original_cwd)
-            
-            if result.returncode == 0:
-                print("Successfully compiled curope using setup.py")
-                return True
-            else:
-                print(f"Setup.py compilation failed: {result.stderr}")
-                return False
-                
-        except Exception as e2:
-            print(f"Alternative method also failed: {e2}")
-            import traceback
-            traceback.print_exc()
-            return False
+subprocess.check_call([
+    sys.executable, "-m", "pip", "install", "-e", 
+    "./extern/CUT3R/src/croco/models/curope"
+])
 
-install_croco()
+import spaces
+import torch
 
 from typing import List, Literal
 # from pathlib import Path