Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -39,7 +39,7 @@ model_paths = [
|
|
39 |
"karths/binary_classification_train_people",
|
40 |
"karths/binary_classification_train_architecture",
|
41 |
]
|
42 |
-
'karths/binary_classification_train_requirement',
|
43 |
quality_mapping = {
|
44 |
'binary_classification_train_test': 'Test',
|
45 |
'binary_classification_train_process': 'Process',
|
@@ -54,7 +54,7 @@ quality_mapping = {
|
|
54 |
'binary_classification_train_architecture':'Architecture'
|
55 |
|
56 |
}
|
57 |
-
'binary_classification_train_requirement': 'Requirement',
|
58 |
|
59 |
# Pre-load models and tokenizer
|
60 |
tokenizer = AutoTokenizer.from_pretrained("distilroberta-base")
|
|
|
39 |
"karths/binary_classification_train_people",
|
40 |
"karths/binary_classification_train_architecture",
|
41 |
]
|
42 |
+
#'karths/binary_classification_train_requirement',
|
43 |
quality_mapping = {
|
44 |
'binary_classification_train_test': 'Test',
|
45 |
'binary_classification_train_process': 'Process',
|
|
|
54 |
'binary_classification_train_architecture':'Architecture'
|
55 |
|
56 |
}
|
57 |
+
#'binary_classification_train_requirement': 'Requirement',
|
58 |
|
59 |
# Pre-load models and tokenizer
|
60 |
tokenizer = AutoTokenizer.from_pretrained("distilroberta-base")
|