CoderHassan commited on
Commit
59b8ff8
·
verified ·
1 Parent(s): 9f9fb7e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -1,6 +1,13 @@
1
  import streamlit as st
2
  from transformers import MarianMTModel, MarianTokenizer
3
 
 
 
 
 
 
 
 
4
  # Load the model and tokenizer
5
  @st.cache_resource
6
  def load_model():
@@ -31,3 +38,4 @@ if st.button("Translate"):
31
  st.markdown(f"### Translated Text:\n{translated_text}")
32
  else:
33
  st.error("Please enter some text to translate.")
 
 
1
  import streamlit as st
2
  from transformers import MarianMTModel, MarianTokenizer
3
 
4
+ # Install SentencePiece if not already installed
5
+ try:
6
+ import sentencepiece
7
+ except ImportError:
8
+ st.error("The SentencePiece library is required but not installed. Please install it using `pip install sentencepiece` and restart the application.")
9
+ st.stop()
10
+
11
  # Load the model and tokenizer
12
  @st.cache_resource
13
  def load_model():
 
38
  st.markdown(f"### Translated Text:\n{translated_text}")
39
  else:
40
  st.error("Please enter some text to translate.")
41
+