Spaces:
Sleeping
Sleeping
Niklauseik
commited on
Commit
·
bd95bf2
1
Parent(s):
97bb6e6
aiya
Browse files
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
|