Update app.py

app.py

@@ -38,13 +38,16 @@ def generate(prompt,history,max_new_tokens,seed):
     buf = ""
     for response in stream:
         output += response.token.text
-        #buf += response.token.text
-        #if response.token.text == "\n" or response.token.text == "</s>" or response.token.text == ". ":
         yield [(prompt,output)], output
     
 def tts(inp,voice,length,noise,width,sen_pause):
+    inp=inp.strip("</s>")
     yield from pp.stream_tts(inp,voice,length,noise,width,sen_pause)
-
+def load_mod(model):
+    yield f"Loading: {model}"
+    pp.load_mod(model)
+    yield f"Voice Loaded: {model}"
+    
 with gr.Blocks() as iface:
     stt=gr.State()
     gr.HTML("""""")
@@ -52,10 +55,11 @@ with gr.Blocks() as iface:
     #chatbot=gr.Chatbot(show_label=False, show_share_button=False, show_copy_button=True, likeable=True, layout="panel"),
     chatbot=gr.Chatbot()
     msg = gr.Textbox()
-    with gr.Row():
-        submit_b = gr.Button()
-        stop_b = gr.Button("Stop")
-        clear = gr.ClearButton([msg, chatbot])
+    with gr.Group():
+        with gr.Row():
+            submit_b = gr.Button()
+            stop_b = gr.Button("Stop")
+            clear = gr.ClearButton([msg, chatbot])
     with gr.Accordion("Voice Controls",open=False):
         names=gr.Dropdown(label="Voice", choices=pp.key_list,value="en_US-joe-medium")
         length=gr.Slider(label="Length", minimum=0.01, maximum=10.0, value=1)
@@ -64,7 +68,7 @@ with gr.Blocks() as iface:
         sen_pause=gr.Slider(label="Sentence Pause", minimum=0.1, maximum=10.0, value=1)
         upd_btn=gr.Button("Update")
 
-    names.change(pp.load_mod,names,submit_b)
+    names.change(load_mod,names,msg)
     sub_b = submit_b.click(generate, [msg,chatbot],[chatbot,stt]).then(tts,[stt,names,length,noise,width,sen_pause],aud)
     sub_e = msg.submit(generate, [msg, chatbot], aud)
     stop_b.click(None,None,None, cancels=[sub_b,sub_e])