Update app.py

app.py

@@ -1,3 +1,4 @@
+
 import os
 import torch
 import time
@@ -15,7 +16,10 @@ from flask_cors import CORS
 import numpy as np
 import trimesh
 import cv2
-from lgm.models import LGM
+try:
+    from lgm.models import LGM
+except ImportError:
+    LGM = None
 
 os.environ["CUDA_VISIBLE_DEVICES"] = ""
 torch.set_default_device("cpu")
@@ -133,6 +137,9 @@ def load_model():
                 else:
                     raise
         
+        if LGM is None:
+            raise ImportError("LGM module not available. Ensure lgm is installed from https://github.com/baaivision/LGM.")
+        
         lgm_model = LGM.from_pretrained(
             model_name,
             cache_dir=CACHE_DIR,
@@ -197,7 +204,7 @@ def progress(job_id):
         
         yield f"data: {json.dumps({'status': 'processing', 'progress': job['progress']})}\n\n"
         
-        last_progress = job['progress']
+’T        last_progress = job['progress']
         check_count = 0
         while job['status'] == 'processing':
             if job['progress'] != last_progress: