lsdrs commited on
Commit
fdb81f3
1 Parent(s): f1df36a

fix: convert env variable to int

Browse files
Files changed (2) hide show
  1. app.py +4 -3
  2. env.example +7 -0
app.py CHANGED
@@ -9,14 +9,15 @@ from utils import (TextGenerationPipeline, from_en_translation,
9
 
10
  load_dotenv()
11
 
12
- max_answer_length = os.getenv('MAX_ANSWER_LENGTH', 100)
13
  desired_language = os.getenv('DESIRED_LANGUAGE', 'pt')
14
  polly_language = os.getenv('POLLY_LANGUAGE_LANGUAGE', 'pt-BR')
 
15
 
16
  response_generator_pipe = TextGenerationPipeline(max_length=max_answer_length)
17
 
18
 
19
- def main(audio: object, use_polly: bool) -> Tuple[str, str, object]:
20
  if audio is None:
21
  return "", "", ""
22
 
@@ -56,7 +57,7 @@ demo = Interface(
56
  fn=main,
57
  inputs=[
58
  Audio(source="microphone", type="filepath"),
59
- Checkbox(value=False, label="Usar Polly?")
60
  ],
61
  outputs=[
62
  Textbox(label="Você disse: "),
 
9
 
10
  load_dotenv()
11
 
12
+ max_answer_length = int(os.getenv('MAX_ANSWER_LENGTH', 100))
13
  desired_language = os.getenv('DESIRED_LANGUAGE', 'pt')
14
  polly_language = os.getenv('POLLY_LANGUAGE_LANGUAGE', 'pt-BR')
15
+ use_polly = os.getenv('USE_POLLY', False)
16
 
17
  response_generator_pipe = TextGenerationPipeline(max_length=max_answer_length)
18
 
19
 
20
+ def main(audio: object, use_polly: bool = use_polly) -> Tuple[str, str, object]:
21
  if audio is None:
22
  return "", "", ""
23
 
 
57
  fn=main,
58
  inputs=[
59
  Audio(source="microphone", type="filepath"),
60
+ Checkbox(value=use_polly, label="Usar Polly?")
61
  ],
62
  outputs=[
63
  Textbox(label="Você disse: "),
env.example ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ DESIRED_LANGUAGE=pt
2
+ POLLY_LANGUAGE=pt-BR
3
+ MAX_ANSWER_LENGTH=100
4
+ AWS_ACCESS_KEY_ID=
5
+ AWS_SECRET_ACCESS_KEY=
6
+ AWS_REGION=
7
+ USE_POLLY=