Spaces:
Sleeping
Sleeping
Niklauseik
commited on
Commit
·
97bb6e6
1
Parent(s):
982df32
aiya
Browse files
app.py
CHANGED
|
@@ -4,11 +4,12 @@ import pandas as pd
|
|
| 4 |
from sklearn.metrics import accuracy_score, precision_recall_fscore_support
|
| 5 |
|
| 6 |
# Define the available models and tasks
|
| 7 |
-
TASKS = ["sentiment-analysis", "text-classification"]
|
| 8 |
MODELS = {
|
| 9 |
"DistilBERT": "distilbert-base-uncased",
|
| 10 |
"BERT": "bert-base-uncased",
|
| 11 |
-
"RoBERTa": "roberta-base"
|
|
|
|
| 12 |
# Add other models here
|
| 13 |
}
|
| 14 |
|
|
|
|
| 4 |
from sklearn.metrics import accuracy_score, precision_recall_fscore_support
|
| 5 |
|
| 6 |
# Define the available models and tasks
|
| 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
|
| 14 |
}
|
| 15 |
|