fix: convert env variable to int
Browse files- app.py +4 -3
- 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=
|
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=
|