debug

app.py

@@ -10,8 +10,15 @@ from transformers import FlaxAutoModelForSeq2SeqLM, AutoTokenizer
 
 @st.cache_resource
 def load_model(model_name, tokenizer_name):
-    model = FlaxAutoModelForSeq2SeqLM.from_pretrained(model_name)
-    tokenizer = AutoTokenizer.from_pretrained(tokenizer_name)
+    try:
+        model = FlaxAutoModelForSeq2SeqLM.from_pretrained(model_name)
+        tokenizer = AutoTokenizer.from_pretrained(tokenizer_name)
+    except Exception as e:
+        st.error(f"Error loading model: {e}")
+        st.error(f"Model not found. Use {DEFAULT_MODEL} instead")
+        model_path = DEFAULT_MODEL
+        model = FlaxAutoModelForSeq2SeqLM.from_pretrained(model_path)
+        tokenizer = AutoTokenizer.from_pretrained(DEFAULT_MODEL)
     return model, tokenizer
 
 
@@ -66,10 +73,6 @@ def main():
 
     model_path = DATASETS_MODEL_INFO[dataset]['languages pairs'][language_pair]['epsilon'][str(epsilon)]
 
-    if not os.path.exists(model_path):
-        st.error(f"Model not found. Use {DEFAULT_MODEL} instead")
-        model_path = DEFAULT_MODEL
-
     model, tokenizer = load_model(model_path, tokenizer_name=DEFAULT_MODEL)
     st.success('Model loaded!')
     st_model_load.text("")

requirements.txt

@@ -8,4 +8,5 @@ torch==1.13.1
 transformers==4.26.0
 streamlit==1.38.0
 optax==0.1.4
-orbax==0.1.1
+orbax==0.1.1
+sentencepiece==0.1.97