Update app.py

app.py

@@ -15,7 +15,7 @@ def format_prompt(message, history):
 def generate(prompt,history)
     seed = random.randint(1,9999999999999)
 
-    system_prompt = agent.SONG_WRITER
+    system_prompt = prompts.SONG_WRITER
     generate_kwargs = dict(
         temperature=0.9,
         max_new_tokens=1024,
@@ -32,4 +32,11 @@ def generate(prompt,history)
     for response in stream:
         output += response.token.text
         yield output
-    return output
+
+
+gr.ChatInterface(
+    fn=generate,
+    chatbot=gr.Chatbot(show_label=False, show_share_button=False, show_copy_button=True, likeable=True, layout="panel"),
+    title="Song Writer",
+    concurrency_limit=20,
+).launch(show_api=False)