Fixes, working version

app.py

@@ -67,9 +67,9 @@ def predict(model, text, melody, duration, topk, topp, temperature, cfg_coef):
         try:
             # Combine the output segments into one long audio file
             output_segments = [segment.detach().cpu().float()[0] for segment in output_segments]
-            output = torch.cat(output_segments, dim=2)
+            output = torch.cat(output_segments, dim=1)
         except Exception as e:
-            print(f"error combining segments: {e}. Using first segment only")
+            print(f"Error combining segments: {e}. Using the first segment only.")
             output = output_segments[0].detach().cpu().float()[0]
     else:
         output = output.detach().cpu().float()[0]

audiocraft/utils/extend.py

@@ -67,7 +67,7 @@ def generate_music_segments(text, melody, MODEL, duration:int=10, segment_durati
         # Append the generated output to the list of segments
         #output_segments.append(output[:, :segment_duration])
         output_segments.append(output)
-        print(f"output_segments: {len(output_segments)}: shape[0]: {output.shape} dim {output.dim()}")
+        print(f"output_segments: {len(output_segments)}: shape: {output.shape} dim {output.dim()}")
     return output_segments
 
 #def generate_music_segments(text, melody, duration, MODEL, segment_duration=30):