test

app.py

@@ -11,7 +11,7 @@ else:
     device = torch.device("cpu")
     logging.warning("GPU not found, using CPU, translation will be very slow.")
 
-st.cache(suppress_st_warning=True, allow_output_mutation=True)
+# st.cache(suppress_st_warning=True, allow_output_mutation=True)
 st.set_page_config(page_title="M2M100 Translator")
 
 lang_id = {
@@ -118,7 +118,7 @@ lang_id = {
 }
 
 
-@st.cache(suppress_st_warning=True, allow_output_mutation=True)
+@st.cache_data(suppress_st_warning=True, allow_output_mutation=True)
 def load_model(
     pretrained_model: str = "facebook/m2m100_1.2B",
     cache_dir: str = "models/",