app.py

app.py

@@ -0,0 +1,20 @@
+pip install transformers
+
+pip install gradio
+
+import torch
+from transformers import AutoModelForCausalLM, AutoTokenizer
+import gradio as gr
+
+model_name = "Song-extender-333"  # Replace with the name of your pre-trained model
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name).cuda()
+
+def extend_music(music_text):
+    input_ids = tokenizer.encode(music_text, return_tensors='pt').cuda()
+    output = model.generate(input_ids, max_length=8*60+50, pad_token_id=tokenizer.pad_token_id)
+    extended_music = tokenizer.decode(output[0], skip_special_tokens=True)
+    return extended_music
+
+iface = gr.Interface(fn=extend_music, inputs="text", outputs="text", title="Music Extender")
+iface.launch()