Update app.py

app.py

@@ -66,8 +66,8 @@ def empty_output_folder(output_dir):
             print(f"Error deleting file {file_path}: {e}")
 
 # Function to create a temporary file with string content
-def create_temp_file(content, prefix, suffix=".txt"):
-    fd, path = tempfile.mkstemp(prefix=prefix, suffix=suffix)
+def create_temp_file(content, suffix=".txt"):
+    fd, path = tempfile.mkstemp(suffix=suffix)
     with os.fdopen(fd, "w", encoding="utf-8") as f:
         f.write(content)
     return path
@@ -94,8 +94,8 @@ def get_last_mp3_file(output_dir):
 
 def infer(genre_txt_content, lyrics_txt_content, num_segments, max_new_tokens):
     # Create temporary files
-    genre_txt_path = create_temp_file(genre_txt_content, prefix="genre_")
-    lyrics_txt_path = create_temp_file(lyrics_txt_content, prefix="lyrics_")
+    genre_txt_path = create_temp_file(genre_txt_content, ".txt")
+    lyrics_txt_path = create_temp_file(lyrics_txt_content, ".txt")
 
     print(f"Genre TXT path: {genre_txt_path}")
     print(f"Lyrics TXT path: {lyrics_txt_path}")