Edward Tang commited on
Commit
d00ecca
·
1 Parent(s): 1139218
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -8,11 +8,17 @@ processor = AutoProcessor.from_pretrained("facebook/hf-seamless-m4t-large")
8
  model = SeamlessM4TModel.from_pretrained("facebook/hf-seamless-m4t-large")
9
 
10
 
11
- text_inputs = processor(text = "Hello, my dog is cute", src_lang="eng", return_tensors="pt")
 
 
 
 
12
 
13
  # from text
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)
16
 
17
  st.write(translated_text_from_text)
18
- st.write("fool me")
 
 
 
8
  model = SeamlessM4TModel.from_pretrained("facebook/hf-seamless-m4t-large")
9
 
10
 
11
+ title = st.text_input('hello', "Hello, my dog is cute")
12
+
13
+
14
+
15
+ text_inputs = processor(text = title, src_lang="eng", return_tensors="pt")
16
 
17
  # from text
18
  output_tokens = model.generate(**text_inputs, tgt_lang="fra", generate_speech=False)
19
  translated_text_from_text = processor.decode(output_tokens[0].tolist()[0], skip_special_tokens=True)
20
 
21
  st.write(translated_text_from_text)
22
+ st.write("fool me")
23
+
24
+