broadfield commited on
Commit
2f14de8
1 Parent(s): 9ee3a27

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ def format_prompt(message, history):
17
  def generate(prompt,history,max_new_tokens,seed):
18
  #seed = random.randint(1,9999999999999)
19
  print(seed)
20
- system_prompt = prompts.SONG_WRITER
21
  generate_kwargs = dict(
22
  temperature=0.9,
23
  max_new_tokens=max_new_tokens,
 
17
  def generate(prompt,history,max_new_tokens,seed):
18
  #seed = random.randint(1,9999999999999)
19
  print(seed)
20
+ system_prompt = prompts.SONG_WRITER.format(history=history)
21
  generate_kwargs = dict(
22
  temperature=0.9,
23
  max_new_tokens=max_new_tokens,