khof312 commited on
Commit
571e8a3
1 Parent(s): e5e9b34

Update requirements.

Browse files
Files changed (4) hide show
  1. moore_app.py +2 -3
  2. requirements.txt +6 -0
  3. src/helpers.py +3 -1
  4. 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
- import pycountry
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
  '''