new commits

app.py

@@ -8,7 +8,6 @@ from utils import *
 def main():
 
     hide_footer()
-    model_list = ["flan-t5-xl"]
     celeb_data = get_celeb_data(f'data.json')
 
     st.sidebar.header("CelebChat")
@@ -78,7 +77,7 @@ def main():
         b64 = st.session_state["celeb_bot"].text_to_speech(autoplay=False)
         md = f"""
         <p>{response}</p>
-        <audio controls autoplay style="display:none;">
+        <audio controls autoplay>
         <source src="data:audio/wav;base64,{b64}" type="audio/wav">
         Your browser does not support the audio element.
         </audio>

celebbot.py

@@ -108,7 +108,7 @@ class CelebBot():
                     knowledge = self.retrieve_knowledge_assertions()
                 else:
                     instruction1 = f'You need to answer the question based on commonsense.'
-                query = f"Context: {instruction1} {knowledge} Question: {self.text} Answer:"
+                query = f"Context: {instruction1} {knowledge}\n\nQuestion: {self.text}\n\nAnswer:"
                 input_ids = self.QA_tokenizer(f"{query}", return_tensors="pt").input_ids
                 outputs = self.QA_model.generate(input_ids, max_length=1024)
                 self.text = self.QA_tokenizer.decode(outputs[0], skip_special_tokens=True)

run_tts.py

@@ -17,6 +17,7 @@ import noisereduce as nr
 import io
 from scipy.io.wavfile import write
 import base64
+import streamlit as st
 
 
 from rtvc.encoder import inference as encoder
@@ -116,7 +117,15 @@ def tts(text, embed_name, nlp, autoplay=True):
     return base64.b64encode(result_bytes).decode()
 
 if __name__ == "__main__":
-    text = "Adkins was raised by a young single mother in various working-class neighbourhoods of London. As a child, she enjoyed singing contemporary pop music and learned to play the guitar and the clarinet. However, it was not until her early teens, when she discovered rhythm-and-blues singer Etta James and other mid-20th-century performers, that she began to consider a musical career. While she honed her talents at a government-funded secondary school for the performing arts, a friend began posting songs Adkins had written and recorded onto the social networking Web site Myspace. Her music eventually caught the attention of record labels, and in 2006, several months after graduating, she signed a contract with XL Recordings."
+    text = "Adkins was raised by a young single mother in various working-class neighbourhoods of London."
     embed_name = "Adele"
     nlp = spacy.load('en_core_web_sm')
-    tts(text, embed_name, nlp)
+    b64 = tts(text, embed_name, nlp, autoplay=False)
+
+    md = f"""
+    <audio controls autoplay>
+    <source src="data:audio/wav;base64,{b64}" type="audio/wav">
+    Your browser does not support the audio element.
+    </audio>
+    """
+    st.markdown(md, unsafe_allow_html=True)