Update app.py

app.py

@@ -32,8 +32,18 @@ from transformers import (
 )
 from huggingface_hub import hf_hub_download
 
-from geofractal.model.vae.vae_lyra_v2 import MultiModalVAE, MultiModalVAEConfig
-LYRA_AVAILABLE = True
+# 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
+
 
 # ============================================================================
 # CONSTANTS
@@ -1274,7 +1284,7 @@ def generate_image(
 def create_demo():
     """Create Gradio interface."""
     
-    with gr.Blocks() as demo:
+    with gr.Blocks(theme="soft") as demo:
         gr.Markdown("""
         # 🌙 Lyra/Lune Flow-Matching Image Generation
         
@@ -1540,4 +1550,4 @@ def create_demo():
 if __name__ == "__main__":
     demo = create_demo()
     demo.queue(max_size=20)
-    demo.launch(show_api=False)
+    demo.launch()