model updates

app.py

@@ -21,12 +21,22 @@ model_checkpoint = "facebook/esm2_t6_8M_UR50D"
 
 tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
 
+# try:
 if model_type == 'Linear T-Cells (MHC Class I Restriction)':
-    model = TFAutoModelForSequenceClassification.from_pretrained('classifier')
+    try:
+        model = TFAutoModelForSequenceClassification.from_pretrained('classifier')
+    except tf.errors.InvalidArgumentError:
+        st.warning("We're experiencing server issues. Please try again later!", icon="⚠️")
 elif model_type == 'Linear T-Cells (MHC Class II Restriction)':
-    model = TFAutoModelForSequenceClassification.from_pretrained('classifier2')
+    try:
+        model = TFAutoModelForSequenceClassification.from_pretrained('classifier2')
+    except tf.errors.InvalidArgumentError:
+        st.warning("We're experiencing server issues. Please try again later!", icon="⚠️")
 elif model_type == 'Linear B-Cell':
-    model = TFAutoModelForSequenceClassification.from_pretrained('bcell')
+    try:
+        model = TFAutoModelForSequenceClassification.from_pretrained('bcell')
+    except tf.errors.InvalidArgumentError:
+        st.warning("We're experiencing server issues. Please try again later!", icon="⚠️")
 # submit button
 if st.button('Submit'):
     locations = []
@@ -58,3 +68,5 @@ if st.button('Submit'):
         return f'background-color: rgb({r}, {g}, {b})'
 
     st.table(locations.style.applymap(color_survived, subset=['Probability']))
+# except InvalidArgumentError:
+#     st.warning("We're experiencing server issues. Please try again later!", icon="⚠️")

bcell/tf_model.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4e1cffc07e744025c7efe0ae8a4e3081fbb5f68f6b9c88b3ccfe0b96979929f1
+oid sha256:fc764936c96b97e3e5678e26e08f8e96eb7a1effaabc4b8cc1173471f0c3eb5d
 size 30211508