Update app.py

app.py

@@ -6,7 +6,7 @@ import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 from diffusers import StableDiffusionPipeline
 from PIL import Image
-from glados_voice import GLaDOSVoice #In testing phase!
+#from glados_voice import GLaDOSVoice #In testing phase!
 import tempfile
 
 
@@ -72,17 +72,17 @@ model = AutoModelForCausalLM.from_pretrained(
     torch_dtype=torch.float16,
     device_map="auto"
 )
-glados_voice = GLaDOSVoice #In testing phase!
+#glados_voice = GLaDOSVoice #In testing phase!
 pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
-def run_chat_with_voice(message, history, files): #In testing phase!
-    history = history or []
-    reply = glados_chat(message, files)
+#def run_chat_with_voice(message, history, files): #In testing phase!
+    #history = history or []
+    #reply = glados_chat(message, files)
     
-    audio_path = glados_voice.generate_speech(reply)
+    #audio_path = glados_voice.generate_speech(reply)
     
-    history.append((message, reply))
-    return history, "", audio_path
+    #history.append((message, reply))
+    #return history, "", audio_path
 
 def extract_text_from_pdfs(files):
     full_text = ""
@@ -189,7 +189,7 @@ with gr.Blocks(css=REMOTE_CSS) as app:
     file_input = gr.File(label="Upload PDF(s)", file_types=[".pdf"], file_count="multiple")
     t2i_prompt = gr.Textbox(placeholder="Or describe an image...", show_label=False)
     generate_btn = gr.Button("🎨 Generate Image")
-    audio_output = gr.Audio(label="GLaDOS Voice") #In Testing Phase!
+    #audio_output = gr.Audio(label="GLaDOS Voice") #In Testing Phase!
 
     def run_chat(message, history, files):
         history = history or []
@@ -205,8 +205,8 @@ with gr.Blocks(css=REMOTE_CSS) as app:
         history.append((prompt, (path,)))
         return history, ""
 
-    #txt.submit(run_chat, [txt, chatbot, file_input], [chatbot, txt])
-    txt.submit(run_chat_with_voice, [txt, chatbot, file_input], [chatbot, txt, audio_output]) #In Testing Phase!
+    txt.submit(run_chat, [txt, chatbot, file_input], [chatbot, txt])
+    #txt.submit(run_chat_with_voice, [txt, chatbot, file_input], [chatbot, txt, audio_output]) #In Testing Phase!
     generate_btn.click(run_generation, [t2i_prompt, chatbot], [chatbot, t2i_prompt])
 
 app.launch()