Commit
·
3fde71f
1
Parent(s):
337d6ca
again
Browse files
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 |
|