update loss

app.py

@@ -236,7 +236,7 @@ with gr.Blocks() as demo:
             with gr.Column(visible=False) as clap_options:
                 clap_target_type = gr.Radio(["Audio", "Text"], label="CLAP Target Type", value="Audio")
                 clap_text_prompt = gr.Textbox(label="CLAP Text Prompt", visible=False)
-                clap_distance_fn = gr.Dropdown([ "cosine", "mse", "l1"], label="CLAP Distance Function", value="cosine")
+                clap_distance_fn = gr.Dropdown(["cosine", "mse", "l1"], label="CLAP Distance Function", value="cosine")
 
     def update_clap_options(loss_function):
         if loss_function == "CLAPFeatureLoss":

inference.py

@@ -97,8 +97,8 @@ class MasteringStyleTransfer:
                     target = reference_tensor
                 else:
                     target = ito_config['clap_text_prompt']
-                losses = self.clap_loss(output_audio, target, self.args.sample_rate, distance_fn=ito_config['clap_distance_fn'])
-                total_loss = losses
+                print(f'ito_config clap_distance_fn: {ito_config["clap_distance_fn"]}')
+                total_loss = self.clap_loss(output_audio, target, self.args.sample_rate, distance_fn=ito_config['clap_distance_fn'])
 
             if total_loss < min_loss:
                 min_loss = total_loss.item()