jbilcke-hf HF staff commited on
Commit
afcc451
1 Parent(s): c29f6b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -101,10 +101,12 @@ You're the storyteller, crafting a short tale for young listeners. Please abide
101
  system_message = os.environ.get("SYSTEM_MESSAGE", default_system_message)
102
  system_message = system_message.replace("CURRENT_DATE", str(datetime.date.today()))
103
 
104
- ROLES = ["Julian","Pirate"]
105
 
106
  ROLE_PROMPTS = {}
107
  ROLE_PROMPTS["Julian"]=system_message
 
 
108
 
109
  #Pirate scenario
110
  character_name= "AI Beard"
@@ -585,8 +587,10 @@ def generate_speech_for_sentence(history, chatbot_role, sentence, return_as_byte
585
  return
586
 
587
  latent_map = {}
 
588
  latent_map["Julian"] = get_latents("voices/julian-bedtime-style-1.wav")
589
  latent_map["Pirate"] = get_latents("voices/pirate_by_coqui.wav")
 
590
 
591
  #### GRADIO INTERFACE ####
592
 
 
101
  system_message = os.environ.get("SYSTEM_MESSAGE", default_system_message)
102
  system_message = system_message.replace("CURRENT_DATE", str(datetime.date.today()))
103
 
104
+ ROLES = ["Cloee","Julian","Pirate","Theranos"]
105
 
106
  ROLE_PROMPTS = {}
107
  ROLE_PROMPTS["Julian"]=system_message
108
+ ROLE_PROMPTS["Cloee"]=system_message
109
+ ROLE_PROMPTS["Theranos"]=system_message
110
 
111
  #Pirate scenario
112
  character_name= "AI Beard"
 
587
  return
588
 
589
  latent_map = {}
590
+ latent_map["Cloee"] = get_latents("voices/cloee-1.wav")
591
  latent_map["Julian"] = get_latents("voices/julian-bedtime-style-1.wav")
592
  latent_map["Pirate"] = get_latents("voices/pirate_by_coqui.wav")
593
+ latent_map["Theranos"] = get_latents("voices/theranos-1.wav")
594
 
595
  #### GRADIO INTERFACE ####
596