Spaces:

broadfield
/

Song-Writer

Runtime error

broadfield commited on Dec 30, 2023

Commit

2f14de8

1 Parent(s): 9ee3a27

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ def format_prompt(message, history):
 def generate(prompt,history,max_new_tokens,seed):
     #seed = random.randint(1,9999999999999)
     print(seed)
-    system_prompt = prompts.SONG_WRITER
     generate_kwargs = dict(
         temperature=0.9,
         max_new_tokens=max_new_tokens,

 def generate(prompt,history,max_new_tokens,seed):
     #seed = random.randint(1,9999999999999)
     print(seed)
+    system_prompt = prompts.SONG_WRITER.format(history=history)
     generate_kwargs = dict(
         temperature=0.9,
         max_new_tokens=max_new_tokens,