update loss

app.py

@@ -243,9 +243,9 @@ with gr.Blocks() as demo:
 
     def update_clap_options(loss_function):
         if loss_function == "CLAPFeatureLoss":
-            return gr.update(visible=True), gr.update(visible=True)
+            return gr.update(visible=False), gr.update(visible=True)
         else:
-            return gr.update(visible=False), gr.update(visible=False)
+            return gr.update(visible=True), gr.update(visible=False)
 
     loss_function.change(
         update_clap_options,
@@ -261,7 +261,7 @@ with gr.Blocks() as demo:
         inputs=[clap_target_type],
         outputs=[clap_text_prompt]
     )
-    
+
     ito_button = gr.Button("Perform ITO")
 
     with gr.Row():

inference.py

@@ -91,13 +91,14 @@ class MasteringStyleTransfer:
             # Compute loss
             if ito_config['loss_function'] == 'AudioFeatureLoss':
                 losses = af_loss(output_audio, reference_tensor)
+                total_loss = sum(losses.values())
             elif ito_config['loss_function'] == 'CLAPFeatureLoss':
                 if ito_config['clap_target_type'] == 'Audio':
                     target = reference_tensor
                 else:
                     target = ito_config['clap_text_prompt']
                 losses = self.clap_loss(output_audio, target, self.args.sample_rate)
-            total_loss = sum(losses.values())
+                total_loss = losses
 
             if total_loss < min_loss:
                 min_loss = total_loss.item()

modules/loss.py

@@ -520,3 +520,4 @@ if __name__ == "__main__":
     loss = clap_loss(input_audio, target_text, sample_rate)
     print(loss)
 
+    print(loss.item())