new commits

app.py

@@ -5,9 +5,9 @@ from utils import *
 
 
 def main():
-
+    st.set_page_config(initial_sidebar_state="expanded")
     hide_footer()
-    model_list = ["flan-t5-xl"]
+    model_list = ["flan-t5-large"]
     celeb_data = get_celeb_data(f'data.json')
 
     st.sidebar.header("CelebChat")

celebbot.py

@@ -111,7 +111,7 @@ class CelebBot():
                     transformed_text.append("my")
                 elif token.text.lower() == name and token.dep_ in ["nsubj", "nsubjpass"]:
                     transformed_text.append("I")
-                elif token.text == "’s" and doc[i-1].text.lower() == name:
+                elif token.text in ["'s", "’s"] and doc[i-1].text.lower() == name:
                     transformed_text[-1] = "my"
                 elif token.text.lower() == "their":
                     transformed_text.append("our")
@@ -129,7 +129,7 @@ class CelebBot():
                         transformed_text.append("me")
                 elif token.text.lower() == name and token.dep_ in ["nsubj", "nsubjpass"]:
                     transformed_text.append("I")
-                elif token.text == "’s" and doc[i-1].text.lower() == name:
+                elif token.text in ["'s", "’s"] and doc[i-1].text.lower() == name:
                     transformed_text[-1] = "my"
                 elif token.text.lower() == "their":
                     transformed_text.append("our")

data.json

@@ -344,6 +344,9 @@
             ], 
             "gender": "M" 
         },
+        "Margot Robbie": {
+            "gender": "F"  
+        }, 
         "Marion Cotillard": {
             "gender": "F"  
         }, 
@@ -356,6 +359,9 @@
         "Michelle Yeoh": {
             "gender": "F"  
         },
+        "Natalie Portman": {
+            "gender": "F"  
+        },
         "Naomi Campbell": {
             "gender": "F"  
         },
@@ -365,12 +371,18 @@
         "Oprah Winfrey": {
             "gender": "F"  
         },
+        "Priyanka Chopra Jonas": {
+            "gender": "F"  
+        },    
         "Penelope Cruz": {
             "gender": "F"  
         },
         "Rihanna": {
             "gender": "F"  
         },
+        "Salma Hayek": {
+            "gender": "F"  
+        },
         "Sandra Bullock": {
             "gender": "F"  
         },

run_tts.py

@@ -71,9 +71,9 @@ def tts(text, embed_name, nlp, autoplay=True):
     ## Save synthesizer visualization results
     if not os.path.exists("syn_results"):
         os.mkdir("syn_results")
-    save_attention_multiple(alignments, "syn_results/attention")
-    save_stop_tokens(stop_tokens, "syn_results/stop_tokens")
-    save_spectrogram(spec, "syn_results/mel")
+    # save_attention_multiple(alignments, "syn_results/attention")
+    # save_stop_tokens(stop_tokens, "syn_results/stop_tokens")
+    # save_spectrogram(spec, "syn_results/mel")
     print("Created the mel spectrogram")