testing

app.py

@@ -1,19 +1,18 @@
 import streamlit as st
-from transformers import AutoProcessor, SeamlessM4Tv2ForTextToText
+from transformers import AutoProcessor, SeamlessM4TModel
 
-st.title("Hot Dog? Or Not!?")
+st.title("Hot Dog? Or Not!!!?")
 
-processor = AutoProcessor.from_pretrained("facebook/seamless-m4t-v2-large")
+processor = AutoProcessor.from_pretrained("facebook/hf-seamless-m4t-large")
 
-model = SeamlessM4Tv2ForTextToText.from_pretrained("facebook/seamless-m4t-v2-large")
+model = SeamlessM4TModel.from_pretrained("facebook/hf-seamless-m4t-large")
 
 
 text_inputs = processor(text = "Hello, my dog is cute", src_lang="eng", return_tensors="pt")
 
 # from text
-output_tokens = model.generate(**text_inputs, tgt_lang="fra")
-translated_text_from_text = processor.decode(output_tokens[0].tolist()[0], skip_special_tokens=True)
-
+output_tokens = model.generate(**text_inputs, tgt_lang="fra", generate_speech=False)
+translated_text_from_text = processor.decode(output_tokens[0].tolist(), skip_special_tokens=True)
 
 st.write(translated_text_from_text)
 st.write("fool me")