Update app.py

app.py

@@ -31,9 +31,9 @@ def load_whisper():
     return whisper.load_model("medium", device  = 'cpu')
 
 def load_tts():
-    tts_model = VitsModel.from_pretrained("facebook/mms-tts-eng")
+    tts_model = VitsModel.from_pretrained("facebook/mms-tts-pl")
     #tts_model.to("cuda")
-    tokenizer_tss = AutoTokenizer.from_pretrained("facebook/mms-tts-eng")
+    tokenizer_tss = AutoTokenizer.from_pretrained("facebook/mms-tts-pl")
     return tts_model, tokenizer_tss
 
 
@@ -109,7 +109,7 @@ def predict(_query, _chatbot, _task_history):
     _chatbot.append((_parse_text(_query), ""))
     full_response = ""
 
-    for response in model.chat_stream(tokenizer, _query, history=_task_history,system = "You are an AI assistant. Please be kind and answer responsibly."  ):
+    for response in model.chat_stream(tokenizer, _query, history=_task_history,system = "Jesteś asystentem AI. Odpowiadaj grzecznie i w języku polskim :)"  ):
         _chatbot[-1] = (_parse_text(_query), _parse_text(response))
 
         yield _chatbot