Update app.py

app.py

@@ -14,7 +14,7 @@ DEFAULT_IMAGE = "original_image.jpg"
 with open(SEMANTIC_DIPOLES_FILE, "r") as f:
     semantic_dipoles = json.load(f)
     # Transform semantic_dipoles into "A -> B" format
-    formatted_dipoles = [f"{pair[0]} -> {pair[1]}" for pair in semantic_dipoles]
+    formatted_dipoles = [f"{pair[1]]} -> {pair[0]}" for pair in semantic_dipoles]
 
 # Helper to list all latent code folders
 latent_code_folders = sorted(
@@ -55,12 +55,15 @@ def display_image(latent_code, formatted_dipole, frame_idx):
 
 # Function to display GAN latent space interactive plot
 def display_interactive_plot(latent_code):
-    file_path = f"files/{LATENT_CODES_DIR}/{latent_code}/interactive_latent_space_{latent_code}.html"
+    # file_path = f"files/{LATENT_CODES_DIR}/{latent_code}/interactive_latent_space_{latent_code}.html"
+    file_path = f"/file/ContraCLIP/experiments/wip/ContraCLIP_stylegan2_ffhq1024-W-K21-D128-lss_beta_0.1-eps0.1_0.2-nonlinear_css_beta_0.5-contrastive_0.5-5000-expressions/results/stylegan2_ffhq1024-4/32_0.2_6.4/{latent_code}/interactive_latent_space_{latent_code}.html"
     iframe_html = f'<iframe src="{file_path}" width="800" height="600" frameborder="0"></iframe>'
     return iframe_html
 
+    
+
 # Gradio Interface
-def build_interface():
+def build_interface(): 
     with gr.Blocks() as demo:
         gr.Markdown("# ContraCLIP-based Image Editing and Visualization Demo")