Update app.py

app.py

@@ -20,7 +20,7 @@ def song_generator(input_prompt):
   # set pad_token_id to eos_token_id because OPT does not have a PAD token
   model.config.pad_token_id = model.config.eos_token_id
   input_ids = tokenizer(input_prompt, return_tensors="pt")
-  stopping_criteria = StoppingCriteriaList([MaxLengthCriteria(max_length=300)])
+  stopping_criteria = StoppingCriteriaList([MaxLengthCriteria(max_length=30)])
   outputs = model.contrastive_search(
       **input_ids, penalty_alpha=0.6, top_k=15, stopping_criteria=stopping_criteria, pad_token_id=tokenizer.eos_token_id,
   )
@@ -35,7 +35,7 @@ submit = st.button('Generate')
 input_song = query
 
 if submit:
-  st.subheader('Song generated')
+  st.subheader('Song generated is ')
   
   with st.spinner(text='This may take a moment...'):
     output_sentence = song_generator(input_song)