Update app.py

app.py

@@ -32,7 +32,7 @@ print('*' * 70)
 print('Loading model and tokenizer...')
 print('*' * 70)
 
-model_name = "asigalov61/Lyrics_Qwen2.5-1.5B-Instruct"
+model_name = "asigalov61/Lyrics_Qwen2.5-0.5B-Instruct"
 
 model = AutoModelForCausalLM.from_pretrained(
     model_name,
@@ -281,7 +281,7 @@ def Morph_Lyrics(input_title, input_lyrics):
     print('=' * 70)
     print('Generating...')
 
-    num_batches = 128
+    num_batches = 256
 
     generated_ids = model.generate(**model_inputs,
                                    max_new_tokens=4096,
@@ -362,7 +362,7 @@ if __name__ == "__main__":
     with app:
         
         gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Lyrics Morpher</h1>")
-        gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Morph any lyrics into an exact variations with fine-tuned Qwen2.5 1.5B Instruct model</h1>")
+        gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Morph any lyrics into an exact variations with fine-tuned Qwen2.5 0.5B Instruct model</h1>")
 
         input_title = gr.Textbox(label="Enter song title here", value="Nothing Else Matters")
         input_lyrics = gr.Textbox(label="Enter song lyrics here", value="So close no matter how far\nCould not be much more from the heart\nForever trusting who we are\nAnd nothing else matters")