rbcurzon commited on
Commit
21ef06a
·
verified ·
1 Parent(s): bbb8c7f

refactor: Add time delay before removing recording audios.

Browse files
Files changed (1) hide show
  1. app.py +10 -2
app.py CHANGED
@@ -30,6 +30,8 @@ from transformers import (
30
  VitsTokenizer,
31
  )
32
 
 
 
33
  import gradio as gr
34
 
35
  import logging
@@ -90,6 +92,11 @@ def translate(text, srcLang, tgtLang):
90
  )
91
  return response.text
92
 
 
 
 
 
 
93
  @app.get("/")
94
  def read_root():
95
  return{
@@ -125,7 +132,7 @@ async def translate_audio(
125
  generate_kwargs=generate_kwargs
126
  )
127
 
128
- print(result)
129
 
130
  result_dict = {
131
  "transcribed_text": result['text'],
@@ -166,6 +173,7 @@ async def translate_text(text: str,
166
  "tgtLang": tgtLang
167
  }
168
  return result_dict
 
169
  @app.post("/synthesize/")
170
  async def synthesize(text: str):
171
  model = VitsModel.from_pretrained("facebook/mms-tts-tgl")
@@ -190,5 +198,5 @@ async def synthesize(text: str):
190
  return FileResponse(
191
  temp_file,
192
  media_type="audio/wav",
193
- background=BackgroundTask(os.remove, temp_file)
194
  )
 
30
  VitsTokenizer,
31
  )
32
 
33
+ import time
34
+
35
  import gradio as gr
36
 
37
  import logging
 
92
  )
93
  return response.text
94
 
95
+ def remove_file(file) {
96
+ time.sleep(600) # delay for 10 minutes
97
+ os.remove(file)
98
+ }
99
+
100
  @app.get("/")
101
  def read_root():
102
  return{
 
132
  generate_kwargs=generate_kwargs
133
  )
134
 
135
+ # print(result)
136
 
137
  result_dict = {
138
  "transcribed_text": result['text'],
 
173
  "tgtLang": tgtLang
174
  }
175
  return result_dict
176
+
177
  @app.post("/synthesize/")
178
  async def synthesize(text: str):
179
  model = VitsModel.from_pretrained("facebook/mms-tts-tgl")
 
198
  return FileResponse(
199
  temp_file,
200
  media_type="audio/wav",
201
+ background=BackgroundTask(remove_file, temp_file)
202
  )