update

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 from transformers import TFAutoModelForSequenceClassification
 from transformers import AutoTokenizer
 import pandas as pd
+import tensorflow as tf
 
 # title
 st.title('Ravens AI')
@@ -25,17 +26,17 @@ tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
 if model_type == 'Linear T-Cells (MHC Class I Restriction)':
     try:
         model = TFAutoModelForSequenceClassification.from_pretrained('classifier')
-    except tf.errors.InvalidArgumentError:
+    except:
         st.warning("We're experiencing server issues. Please try again later!", icon="⚠️")
 elif model_type == 'Linear T-Cells (MHC Class II Restriction)':
     try:
         model = TFAutoModelForSequenceClassification.from_pretrained('classifier2')
-    except tf.errors.InvalidArgumentError:
+    except:
         st.warning("We're experiencing server issues. Please try again later!", icon="⚠️")
 elif model_type == 'Linear B-Cell':
     try:
         model = TFAutoModelForSequenceClassification.from_pretrained('bcell')
-    except tf.errors.InvalidArgumentError:
+    except:
         st.warning("We're experiencing server issues. Please try again later!", icon="⚠️")
 # submit button
 if st.button('Submit'):