Niklauseik commited on
Commit
97bb6e6
·
1 Parent(s): 982df32
Files changed (1) hide show
  1. app.py +3 -2
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