Edward Tang commited on
Commit
ab26a20
·
1 Parent(s): b34763a
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -1,15 +1,18 @@
1
  import streamlit as st
2
  from transformers import AutoProcessor, SeamlessM4TModel
3
 
4
- st.title("Ed's Awkward Hot Dog? Or Not!!!!!?")
5
 
6
  processor = AutoProcessor.from_pretrained("facebook/hf-seamless-m4t-large")
7
 
8
  model = SeamlessM4TModel.from_pretrained("facebook/hf-seamless-m4t-large")
9
 
 
 
 
10
  def submit():
11
  st.write('method')
12
- mytitle = st.session_state.title
13
  text_inputs = processor(text = mytitle, src_lang="eng", return_tensors="pt")
14
  output_tokens = model.generate(**text_inputs, tgt_lang="fra", generate_speech=False)
15
  translated_text_from_text = processor.decode(output_tokens[0].tolist()[0], skip_special_tokens=True)
 
1
  import streamlit as st
2
  from transformers import AutoProcessor, SeamlessM4TModel
3
 
4
+ st.title("Ed's new Hot Dog? Or Not!!!!!?")
5
 
6
  processor = AutoProcessor.from_pretrained("facebook/hf-seamless-m4t-large")
7
 
8
  model = SeamlessM4TModel.from_pretrained("facebook/hf-seamless-m4t-large")
9
 
10
+ if "texttotranslate" not in st.session_state:
11
+ st.session_state.texttotranslate = ""
12
+
13
  def submit():
14
  st.write('method')
15
+ mytitle = st.session_state.texttotranslate
16
  text_inputs = processor(text = mytitle, src_lang="eng", return_tensors="pt")
17
  output_tokens = model.generate(**text_inputs, tgt_lang="fra", generate_speech=False)
18
  translated_text_from_text = processor.decode(output_tokens[0].tolist()[0], skip_special_tokens=True)