vishwask commited on
Commit
36538f6
1 Parent(s): 24f068c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -27,6 +27,13 @@ from pydub import AudioSegment
27
  from streamlit_extras.streaming_write import write
28
  import time
29
 
 
 
 
 
 
 
 
30
  user_session_id = uuid.uuid4()
31
 
32
  logger = get_logger(__name__)
 
27
  from streamlit_extras.streaming_write import write
28
  import time
29
 
30
+ import transformers
31
+ from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
32
+ translation_model = MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")
33
+ translation_tokenizer = MBart50TokenizerFast.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")
34
+
35
+
36
+
37
  user_session_id = uuid.uuid4()
38
 
39
  logger = get_logger(__name__)