Add code

app.py

@@ -207,20 +207,24 @@ def generate_base(subject, setting):
                                               "a minute to read")},
                 {"role": "user", "content": f"Please tell me a story about a {subject} in {setting}"}]
     gr.Info("Generating story", duration=3)
-    response = client.chat_completion(messages, max_tokens=2048, seed=random.randint(1, 5000))
+    response = client.chat_completion(messages, max_tokens=1024, seed=random.randint(1, 5000))
     gr.Info("Story Generated", duration=3)
     story = response.choices[0].message.content
 
     model_input = story.replace("\n", " ").strip()
     model_input_tokens = nltk.sent_tokenize(model_input)
+    model_input_token_pairs = []
+    for i in range(0, len(model_input_tokens), 2):
+        model_input_token_pairs.append(" ".join(model_input_tokens[i:(i + 2)]))
 
-    play_steps_in_s = 4.0
+
+    play_steps_in_s = 5.0
     play_steps = int(frame_rate * play_steps_in_s)
 
     description = "Jenny speaks at an average pace with a calm delivery in a very confined sounding environment with clear audio quality."
     description_tokens = tokenizer(description, return_tensors="pt").to(device)
 
-    for i, sentence in enumerate(model_input):
+    for i, sentence in enumerate(model_input_token_pairs):
         streamer = ParlerTTSStreamer(model, device=device, play_steps=play_steps)
 
         prompt = tokenizer(sentence, return_tensors="pt").to(device)
@@ -253,8 +257,8 @@ with gr.Blocks() as block:
     )
     with gr.Group():
         with gr.Row():
-            subject = gr.Dropdown(value="Princess", choices=["Prince", "Princess", "Dog", "Cat"])
-            setting = gr.Dropdown(value="Forest", choices=["Forest", "Kingdom", "Jungle", "Underwater"])
+            subject = gr.Dropdown(value="Princess", choices=["Prince", "Princess", "Dog", "Cat"], label="Subject")
+            setting = gr.Dropdown(value="Forest", choices=["Forest", "Kingdom", "Jungle", "Underwater", "Pirate Ship"], label="Setting")
         with gr.Row():
             run_button = gr.Button("Generate Story", variant="primary")
     with gr.Row():