Update requirements.
Browse files- moore_app.py +2 -3
- requirements.txt +6 -0
- src/helpers.py +3 -1
- src/text_to_speech.py +0 -1
moore_app.py
CHANGED
@@ -5,11 +5,10 @@ import streamlit as st
|
|
5 |
from transformers import set_seed, pipeline
|
6 |
from transformers import VitsTokenizer, VitsModel
|
7 |
from datasets import load_dataset, Audio
|
8 |
-
from IPython.display import Audio as Aud
|
9 |
from src import *
|
10 |
|
11 |
-
from huggingface_hub import login
|
12 |
-
from dotenv import load_dotenv
|
13 |
|
14 |
#load_dotenv()
|
15 |
#HUGGINGFACE_KEY = os.environ.get("HUGGINGFACE_KEY")
|
|
|
5 |
from transformers import set_seed, pipeline
|
6 |
from transformers import VitsTokenizer, VitsModel
|
7 |
from datasets import load_dataset, Audio
|
|
|
8 |
from src import *
|
9 |
|
10 |
+
#from huggingface_hub import login
|
11 |
+
#from dotenv import load_dotenv
|
12 |
|
13 |
#load_dotenv()
|
14 |
#HUGGINGFACE_KEY = os.environ.get("HUGGINGFACE_KEY")
|
requirements.txt
ADDED
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
datasets
|
2 |
+
librosa
|
3 |
+
pycountry
|
4 |
+
scipy
|
5 |
+
transformers
|
6 |
+
torch
|
src/helpers.py
CHANGED
@@ -1,3 +1,5 @@
|
|
|
|
|
|
1 |
|
2 |
iso_encoder = {"English":"eng",
|
3 |
"French":"fra",
|
@@ -5,7 +7,7 @@ iso_encoder = {"English":"eng",
|
|
5 |
|
6 |
iso_decoder = dict((v,k) for k,v in iso_encoder.items())
|
7 |
|
8 |
-
|
9 |
|
10 |
def encode_iso(lang:str)-> str:
|
11 |
''' Takes the name of a language and returns its ISO-3 code. '''
|
|
|
1 |
+
import pycountry
|
2 |
+
|
3 |
|
4 |
iso_encoder = {"English":"eng",
|
5 |
"French":"fra",
|
|
|
7 |
|
8 |
iso_decoder = dict((v,k) for k,v in iso_encoder.items())
|
9 |
|
10 |
+
|
11 |
|
12 |
def encode_iso(lang:str)-> str:
|
13 |
''' Takes the name of a language and returns its ISO-3 code. '''
|
src/text_to_speech.py
CHANGED
@@ -2,7 +2,6 @@
|
|
2 |
import torch
|
3 |
from transformers import set_seed
|
4 |
from transformers import VitsTokenizer, VitsModel
|
5 |
-
from IPython.display import Audio as Aud
|
6 |
|
7 |
def synthesize_facebook(s:str, iso3:str) -> str:
|
8 |
'''
|
|
|
2 |
import torch
|
3 |
from transformers import set_seed
|
4 |
from transformers import VitsTokenizer, VitsModel
|
|
|
5 |
|
6 |
def synthesize_facebook(s:str, iso3:str) -> str:
|
7 |
'''
|