Update app.py

app.py

@@ -13,8 +13,8 @@ title1 = "RWKV-4-Raven-7B-v8-Eng-20230408-ctx4096"
 os.environ["RWKV_JIT_ON"] = '1'
 os.environ["RWKV_CUDA_ON"] = '1' # if '1' then use CUDA kernel for seq mode (much faster)
 
-#from TTS.api import TTS
-#tts = TTS(model_name="tts_models/multilingual/multi-dataset/your_tts", progress_bar=False, gpu=True)
+from TTS.api import TTS
+tts = TTS(model_name="tts_models/multilingual/multi-dataset/your_tts", progress_bar=False, gpu=True)
 
 from rwkv.model import RWKV
 model_path = hf_hub_download(repo_id="BlinkDL/rwkv-4-raven", filename=f"{title1}.pth")
@@ -41,7 +41,7 @@ def generate_prompt(instruction, input=None):
 def evaluate(
 #    instruction,
     audio,
-#    upload,
+    upload,
     input=None,
     token_count=200,
     temperature=1.0,
@@ -109,11 +109,13 @@ def evaluate(
 
 #    res1 = ' '.join(str(x) for x in res)
     
-#    tts.tts_to_file(res1, speaker_wav = upload, language="en", file_path="output.wav")
+    tts.tts_to_file(res1, speaker_wav = upload, language="en", file_path="output.wav")
     
-    return out_str.strip()
+#    return out_str.strip()
     
 #    return [result.text, res]    
+
+    return "output.wav"
     
 #    yield out_str.strip()
 
@@ -131,11 +133,11 @@ g = gr.Interface(
         gr.components.Slider(0.0, 1.0, step=0.1, value=0.4),  # countPenalty        
     ],
     outputs=[
-        gr.inputs.Textbox(
-            lines=5,
-            label="Raven Output",
-        ),
-#        gr.Audio(label="Audio with Custom Voice"),
+#        gr.inputs.Textbox(
+#            lines=5,
+#            label="Raven Output",
+#        ),
+        gr.Audio(label="Audio with Custom Voice"),
     ],
     title="🥳💬💕 - TalktoAI，随时随地，谈天说地！",
     description="🤖 - 让有人文关怀的AI造福每一个人！AI向善，文明璀璨！TalktoAI - Enable the future！",