Niklauseik commited on
Commit
bd95bf2
·
1 Parent(s): 97bb6e6
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from sklearn.metrics import accuracy_score, precision_recall_fscore_support
7
  TASKS = ["sentiment-analysis", "ner", "text-classification"]
8
  MODELS = {
9
  "DistilBERT": "distilbert-base-uncased",
10
- "BERT": "bert-base-uncased",
11
  "RoBERTa": "roberta-base",
12
  "LLaMA": "decapoda-research/llama-7b-hf"
13
  # Add other models here
 
7
  TASKS = ["sentiment-analysis", "ner", "text-classification"]
8
  MODELS = {
9
  "DistilBERT": "distilbert-base-uncased",
10
+ "BERT": "google-bert/bert-base-uncased",
11
  "RoBERTa": "roberta-base",
12
  "LLaMA": "decapoda-research/llama-7b-hf"
13
  # Add other models here