Abbas133 commited on
Commit
7e0701a
·
verified ·
1 Parent(s): ee683a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -2,7 +2,7 @@ import streamlit as st
2
  from transformers import M2M100ForConditionalGeneration, M2M100Tokenizer
3
 
4
  # Load the model and tokenizer once at the start
5
- @st.cache_resource(allow_output_mutation=True)
6
  def load_model_and_tokenizer():
7
  model_name = "facebook/m2m100_418M"
8
  tokenizer = M2M100Tokenizer.from_pretrained(model_name)
@@ -55,4 +55,3 @@ if st.button("Translate"):
55
  st.write(translated_text)
56
  else:
57
  st.write("Please enter text to translate.")
58
-
 
2
  from transformers import M2M100ForConditionalGeneration, M2M100Tokenizer
3
 
4
  # Load the model and tokenizer once at the start
5
+ @st.cache_resource
6
  def load_model_and_tokenizer():
7
  model_name = "facebook/m2m100_418M"
8
  tokenizer = M2M100Tokenizer.from_pretrained(model_name)
 
55
  st.write(translated_text)
56
  else:
57
  st.write("Please enter text to translate.")