Spaces:
Sleeping
Sleeping
Nazarshia2889
commited on
Commit
•
6b98882
1
Parent(s):
523e667
update
Browse files
app.py
CHANGED
@@ -2,6 +2,7 @@ import streamlit as st
|
|
2 |
from transformers import TFAutoModelForSequenceClassification
|
3 |
from transformers import AutoTokenizer
|
4 |
import pandas as pd
|
|
|
5 |
|
6 |
# title
|
7 |
st.title('Ravens AI')
|
@@ -25,17 +26,17 @@ tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
|
|
25 |
if model_type == 'Linear T-Cells (MHC Class I Restriction)':
|
26 |
try:
|
27 |
model = TFAutoModelForSequenceClassification.from_pretrained('classifier')
|
28 |
-
except
|
29 |
st.warning("We're experiencing server issues. Please try again later!", icon="⚠️")
|
30 |
elif model_type == 'Linear T-Cells (MHC Class II Restriction)':
|
31 |
try:
|
32 |
model = TFAutoModelForSequenceClassification.from_pretrained('classifier2')
|
33 |
-
except
|
34 |
st.warning("We're experiencing server issues. Please try again later!", icon="⚠️")
|
35 |
elif model_type == 'Linear B-Cell':
|
36 |
try:
|
37 |
model = TFAutoModelForSequenceClassification.from_pretrained('bcell')
|
38 |
-
except
|
39 |
st.warning("We're experiencing server issues. Please try again later!", icon="⚠️")
|
40 |
# submit button
|
41 |
if st.button('Submit'):
|
|
|
2 |
from transformers import TFAutoModelForSequenceClassification
|
3 |
from transformers import AutoTokenizer
|
4 |
import pandas as pd
|
5 |
+
import tensorflow as tf
|
6 |
|
7 |
# title
|
8 |
st.title('Ravens AI')
|
|
|
26 |
if model_type == 'Linear T-Cells (MHC Class I Restriction)':
|
27 |
try:
|
28 |
model = TFAutoModelForSequenceClassification.from_pretrained('classifier')
|
29 |
+
except:
|
30 |
st.warning("We're experiencing server issues. Please try again later!", icon="⚠️")
|
31 |
elif model_type == 'Linear T-Cells (MHC Class II Restriction)':
|
32 |
try:
|
33 |
model = TFAutoModelForSequenceClassification.from_pretrained('classifier2')
|
34 |
+
except:
|
35 |
st.warning("We're experiencing server issues. Please try again later!", icon="⚠️")
|
36 |
elif model_type == 'Linear B-Cell':
|
37 |
try:
|
38 |
model = TFAutoModelForSequenceClassification.from_pretrained('bcell')
|
39 |
+
except:
|
40 |
st.warning("We're experiencing server issues. Please try again later!", icon="⚠️")
|
41 |
# submit button
|
42 |
if st.button('Submit'):
|