Update app.py

app.py

@@ -240,7 +240,7 @@ if st.button("Run"):
     if model is not None:
         with st.spinner(text=f"Generating lyrics..."):
             encoded_prompt = tokenizer(start, add_special_tokens=False, return_tensors="pt").input_ids
-            encoded_prompt = encoded_prompt.to(trainer.model.device)
+            encoded_prompt = encoded_prompt.to(model.device)
             # prediction
             output_sequences = model.generate(
                                     input_ids=encoded_prompt,