rajistics commited on
Commit
b968ce4
1 Parent(s): 9b041cc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -18
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