Update inference/infer.py

inference/infer.py

@@ -121,27 +121,17 @@ stage1_output_set = []
 # genre tags support instrumental，genre，mood，vocal timbr and vocal gender
 # all kinds of tags are needed
 # Ensure files exist
-# Read genre file with UTF-8 encoding
-with open(args.genre_txt, encoding="utf-8") as f:
+with open(args.genre_txt) as f:
     genres = f.read().strip()
-
-# Read lyrics file with UTF-8 encoding
-with open(args.lyrics_txt, encoding="utf-8") as f:
-    raw_lyrics = f.read()
-
-# Format lyrics content and ensure proper newline spacing between blocks
-formatted_lyrics = "\n\n".join([block.strip() for block in raw_lyrics.split("\n\n")])
-
-# Now, split the formatted lyrics into blocks
-lyrics = split_lyrics(formatted_lyrics)
-
-# Join all lyrics into one single string for the full lyrics
+    print(genres)
+with open(args.lyrics_txt) as f:
+    lyrics = split_lyrics(f.read())
+    print(lyrics)
+# intruction
 full_lyrics = "\n".join(lyrics)
-
-# Prepare the prompt with genre and full lyrics
 prompt_texts = [f"Generate music from the given lyrics segment by segment.\n[Genre] {genres}\n{full_lyrics}"]
 prompt_texts += lyrics
-
+print(prompt_texts)
 
 random_id = uuid.uuid4()
 output_seq = None