Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -45,24 +45,6 @@ st.info(
|
|
45 |
"Due to limited resources only low memory models are available. Run this [app locally](https://github.com/cdpierse/transformers-interpret-streamlit) to run the full selection of available models. "
|
46 |
)
|
47 |
|
48 |
-
# uncomment the options below to test out the app with a variety of classification models.
|
49 |
-
models = {
|
50 |
-
# "textattack/distilbert-base-uncased-rotten-tomatoes": "",
|
51 |
-
# "textattack/bert-base-uncased-rotten-tomatoes": "",
|
52 |
-
# "textattack/roberta-base-rotten-tomatoes": "",
|
53 |
-
# "mrm8488/bert-mini-finetuned-age_news-classification": "BERT-Mini finetuned on AG News dataset. Predicts news class (sports/tech/business/world) of text.",
|
54 |
-
# "nateraw/bert-base-uncased-ag-news": "BERT finetuned on AG News dataset. Predicts news class (sports/tech/business/world) of text.",
|
55 |
-
"distilbert-base-uncased-finetuned-sst-2-english": "DistilBERT model finetuned on SST-2 sentiment analysis task. Predicts positive/negative sentiment.",
|
56 |
-
# "ProsusAI/finbert": "BERT model finetuned to predict sentiment of financial text. Finetuned on Financial PhraseBank data. Predicts positive/negative/neutral.",
|
57 |
-
"sampathkethineedi/industry-classification": "DistilBERT Model to classify a business description into one of 62 industry tags.",
|
58 |
-
"MoritzLaurer/policy-distilbert-7d": "DistilBERT model finetuned to classify text into one of seven political categories.",
|
59 |
-
# # "MoritzLaurer/covid-policy-roberta-21": "(Under active development ) RoBERTA model finetuned to identify COVID policy measure classes ",
|
60 |
-
# "mrm8488/bert-tiny-finetuned-sms-spam-detection": "Tiny bert model finetuned for spam detection. 0 == not spam, 1 == spam",
|
61 |
-
}
|
62 |
-
model_name = st.sidebar.selectbox(
|
63 |
-
"Choose a classification model", list(models.keys())
|
64 |
-
)
|
65 |
-
model, tokenizer = load_model(model_name)
|
66 |
|
67 |
|
68 |
|
|
|
45 |
"Due to limited resources only low memory models are available. Run this [app locally](https://github.com/cdpierse/transformers-interpret-streamlit) to run the full selection of available models. "
|
46 |
)
|
47 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
48 |
|
49 |
|
50 |
|