Update app.py

app.py

@@ -33,16 +33,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.model.vae.vae_lyra_v2 import MultiModalVAE, MultiModalVAEConfig
 
 
 # ============================================================================