Update app.py

app.py

@@ -14,8 +14,8 @@ import numpy as np
 from TTS.utils.manage import ModelManager
 from TTS.utils.synthesizer import Synthesizer
 
-emotion_tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-emotion")
-emotion_model = AutoModelWithLMHead.from_pretrained("mrm8488/t5-base-finetuned-emotion")
+#emotion_tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-emotion")
+#emotion_model = AutoModelWithLMHead.from_pretrained("mrm8488/t5-base-finetuned-emotion")
 
 def get_emotion(text):
   input_ids = tokenizer.encode(text + '</s>', return_tensors='pt')
@@ -131,7 +131,7 @@ def greet(character,your_voice,message,history):
 
   history["message_history"].append((message, response))
 
-  emotion = get_emotion(response)
+  #emotion = get_emotion(response)
   
   html = "<div class='chatbot'>"
   for user_msg, resp_msg in history["message_history"]:
@@ -139,7 +139,7 @@ def greet(character,your_voice,message,history):
       html += f"<div class='resp_msg'>{character}: {resp_msg}</div>"
   html += "</div>"
 
-  return "tts_output.wav",emotion, html,history
+  return "tts_output.wav", html,history
   
   
 
@@ -169,7 +169,7 @@ examples=[['Gandalf','Hello','dragon.wav']]
 history =   {"character": "None", "message_history" : [] }
 interface= gr.Interface(fn=greet, 
                         inputs=[gr.inputs.Dropdown(personality_choices),gr.inputs.Audio(source="microphone", type="filepath") ,"text", "state"], 
-                        outputs=[gr.outputs.Audio(type="file"),"text","html","state"],      
+                        outputs=[gr.outputs.Audio(type="file"),"html","state"],      
                         css=css, title=title, description=description,article=article )