shamvideo commited on
Commit
05b5e18
·
verified ·
1 Parent(s): a05431a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -5
app.py CHANGED
@@ -1,3 +1,8 @@
 
 
 
 
 
1
  from transformers import pipeline
2
  import speech_recognition as sr
3
  from gtts import gTTS
@@ -22,19 +27,19 @@ def recognize_speech():
22
  print("Вы сказали: " + text)
23
  return text
24
  except sr.UnknownValueError:
25
- print("Не удалось распознать речь")
26
  return ""
27
  except sr.RequestError as e:
28
  print("Ошибка сервиса распознавания речи; {0}".format(e))
29
  return ""
30
 
31
- # Функция для синтеза речи
32
  def synthesize_speech(text):
33
  tts = gTTS(text=text, lang="ru")
34
  tts.save("response.mp3")
35
  os.system("mpg321 response.mp3")
36
 
37
- # Функция для загрузки документов с Google Drive
38
  def load_documents_from_drive():
39
  creds = Credentials.from_authorized_user_file('token.json', ['https://www.googleapis.com/auth/drive.readonly'])
40
  service = build('drive', 'v3', credentials=creds)
@@ -54,12 +59,12 @@ def load_documents_from_drive():
54
  documents.append(item['name'])
55
  return documents
56
 
57
- # Основной цикл чатбота
58
  def main():
59
  while True:
60
  user_input = recognize_speech()
61
  if user_input:
62
- response = chatbot(user_input, max_length=100, num_return_sequences=1)[0]['generated_text']
63
  print("Чатбот: " + response)
64
  synthesize_speech(response)
65
 
 
1
+ import os
2
+
3
+ # Настройка переменной окружения
4
+ os.environ['TRANSFORMERS_CACHE'] = '/path/to/writable/directory'
5
+
6
  from transformers import pipeline
7
  import speech_recognition as sr
8
  from gtts import gTTS
 
27
  print("Вы сказали: " + text)
28
  return text
29
  except sr.UnknownValueError:
30
+ print("Не удалось распознавать речь")
31
  return ""
32
  except sr.RequestError as e:
33
  print("Ошибка сервиса распознавания речи; {0}".format(e))
34
  return ""
35
 
36
+ # Функция для синтезиса речи
37
  def synthesize_speech(text):
38
  tts = gTTS(text=text, lang="ru")
39
  tts.save("response.mp3")
40
  os.system("mpg321 response.mp3")
41
 
42
+ # Функция для загрузкии документов с Google Drive
43
  def load_documents_from_drive():
44
  creds = Credentials.from_authorized_user_file('token.json', ['https://www.googleapis.com/auth/drive.readonly'])
45
  service = build('drive', 'v3', credentials=creds)
 
59
  documents.append(item['name'])
60
  return documents
61
 
62
+ # Основой цикл чатбота
63
  def main():
64
  while True:
65
  user_input = recognize_speech()
66
  if user_input:
67
+ response = chatbot(user_input, max_length=100, numreturn_sequences=1)[0]['generated_text']
68
  print("Чатбот: " + response)
69
  synthesize_speech(response)
70