Update app.py

app.py

@@ -32,17 +32,8 @@ from transformers import (
 )
 from huggingface_hub import hf_hub_download
 
-# Import Lyra VAE from geofractal
-try:
-    from geofractal.models.vae.vae_lyra import MultiModalVAE, MultiModalVAEConfig
-    LYRA_AVAILABLE = True
-except ImportError:
-    try:
-        from geofractal.train.model.vae.vae_lyra import MultiModalVAE, MultiModalVAEConfig
-        LYRA_AVAILABLE = True
-    except ImportError:
-        print("⚠️ Lyra VAE not available - install geofractal")
-        LYRA_AVAILABLE = False
+from geofractal.models.vae.vae_lyra_v2 import MultiModalVAE, MultiModalVAEConfig
+LYRA_AVAILABLE = True
 
 
 # ============================================================================