chaseharmon commited on
Commit
3fde71f
·
1 Parent(s): 337d6ca
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,6 +46,7 @@ st.header("Supported Artists")
46
  st.write("Lupe Fiasco, Common, Jay-Z, Yasiin Bey, Ab-Soul, Rakim")
47
 
48
  display = "Ask Rap-Mistral Something"
 
49
  question = st.chat_input("Write a verse in the style of Lupe Fiasco")
50
  if question:
51
  prompt = build_prompt(question)
@@ -53,5 +54,4 @@ if question:
53
  model_inputs = inputs.to('cuda')
54
  generated_ids = model.generate(**model_inputs, max_new_tokens=300, do_sample=True, pad_token_id=tokenizer.eos_token_id)
55
  display = tokenizer.batch_decode(generated_ids)
56
- st.write(display)
57
 
 
46
  st.write("Lupe Fiasco, Common, Jay-Z, Yasiin Bey, Ab-Soul, Rakim")
47
 
48
  display = "Ask Rap-Mistral Something"
49
+ st.write(display)
50
  question = st.chat_input("Write a verse in the style of Lupe Fiasco")
51
  if question:
52
  prompt = build_prompt(question)
 
54
  model_inputs = inputs.to('cuda')
55
  generated_ids = model.generate(**model_inputs, max_new_tokens=300, do_sample=True, pad_token_id=tokenizer.eos_token_id)
56
  display = tokenizer.batch_decode(generated_ids)
 
57