Add Python 3.10, fix dependencies, add detailed logging

README.md

@@ -8,6 +8,7 @@ sdk_version: 4.44.0
 app_file: app.py
 pinned: false
 license: cc-by-nc-sa-4.0
+python_version: 3.10
 ---
 
 # Multi-View 3D Reconstruction (MV3DR)

app.py

@@ -2,30 +2,51 @@ import os
 import sys
 import torch
 import subprocess
+import logging
+
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+logger = logging.getLogger(__name__)
+
+logger.info("="*50)
+logger.info("Starting MV3DR Application")
+logger.info("="*50)
 
 if not os.path.exists('dust3r/.git'):
-    print("Initializing dust3r submodule...")
+    logger.info("Initializing dust3r submodule...")
     subprocess.run(['git', 'submodule', 'update', '--init', '--recursive'], check=False)
 
+logger.info("Adding dust3r to Python path...")
 sys.path.append(os.path.join(os.path.dirname(__file__), 'dust3r'))
 
+logger.info("Importing model initialization module...")
 from model import initialize
+
+logger.info("Importing UI builder...")
 from gradio_ui import build_ui
+
+logger.info("Importing configuration...")
 from config import WEIGHTS_PATH, OUTPUT_DIR, SERVER_NAME, SERVER_PORT, SHARE, SHOW_ERROR
 
 
 def main():
+    logger.info("Creating output directory...")
     os.makedirs(OUTPUT_DIR, exist_ok=True)
     
     device = "cuda" if torch.cuda.is_available() else "cpu"
-    print(f"Using device: {device}")
+    logger.info(f"Using device: {device}")
     
-    print("Loading DUSt3R model...")
+    logger.info("Loading DUSt3R model...")
+    logger.info(f"Model weights path: {WEIGHTS_PATH}")
     model = initialize(WEIGHTS_PATH, device)
+    logger.info("Model loaded successfully!")
     
-    print("Starting Multi-View 3D Reconstruction (MV3DR)...")
+    logger.info("Building Gradio UI...")
     app = build_ui(OUTPUT_DIR, model, device)
+    logger.info("UI built successfully!")
+    
+    logger.info("Launching application...")
     app.launch()
+    logger.info("Application started!")
 
 
 if __name__ == "__main__":

model.py

@@ -1,11 +1,17 @@
 import torch
-from dust3r.model import AsymmetricCroCo3DStereo, inf
+import logging
+from dust3r.model import AsymmetricCroCo3DStereo
+
+logger = logging.getLogger(__name__)
 
 
 def initialize(model_path: str, device: str) -> torch.nn.Module:
-    print(f"Loading model from: {model_path}")
+    logger.info(f"Loading model from: {model_path}")
+    
+    logger.info("Loading checkpoint...")
     ckpt = torch.load(model_path, map_location='cpu', weights_only=False)
     
+    logger.info("Parsing model arguments...")
     args = ckpt['args'].model.replace("ManyAR_PatchEmbed", "PatchEmbedDust3R")
 
     if 'landscape_only' not in args:
@@ -13,8 +19,14 @@ def initialize(model_path: str, device: str) -> torch.nn.Module:
     else:
         args = args.replace(" ", "").replace('landscape_only=True', 'landscape_only=False')
 
+    logger.info("Instantiating model...")
     net = eval(args)
+    
+    logger.info("Loading model weights...")
     net.load_state_dict(ckpt['model'], strict=False)
     
-    print(f"Model loaded successfully on {device}")
-    return net.to(device)
+    logger.info(f"Moving model to {device}...")
+    model = net.to(device)
+    
+    logger.info("Model initialization complete!")
+    return model

requirements.txt

@@ -11,3 +11,5 @@ pillow
 tqdm
 numpy
 opencv-python
+pydub
+ffmpeg-python