Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -27,8 +27,7 @@ login(token)
|
|
| 27 |
|
| 28 |
# Model paths and quality mapping
|
| 29 |
model_paths = [
|
| 30 |
-
'karths/binary_classification_train_test',
|
| 31 |
-
'karths/binary_classification_train_requirement',
|
| 32 |
"karths/binary_classification_train_process",
|
| 33 |
"karths/binary_classification_train_infrastructure",
|
| 34 |
"karths/binary_classification_train_documentation",
|
|
@@ -40,10 +39,9 @@ model_paths = [
|
|
| 40 |
"karths/binary_classification_train_people",
|
| 41 |
"karths/binary_classification_train_architecture",
|
| 42 |
]
|
| 43 |
-
|
| 44 |
quality_mapping = {
|
| 45 |
'binary_classification_train_test': 'Test',
|
| 46 |
-
'binary_classification_train_requirement': 'Requirement',
|
| 47 |
'binary_classification_train_process': 'Process',
|
| 48 |
'binary_classification_train_infrastructure': 'Infrastructure',
|
| 49 |
'binary_classification_train_documentation': 'Documentation',
|
|
@@ -56,6 +54,7 @@ quality_mapping = {
|
|
| 56 |
'binary_classification_train_architecture':'Architecture'
|
| 57 |
|
| 58 |
}
|
|
|
|
| 59 |
|
| 60 |
# Pre-load models and tokenizer
|
| 61 |
tokenizer = AutoTokenizer.from_pretrained("distilroberta-base")
|
|
|
|
| 27 |
|
| 28 |
# Model paths and quality mapping
|
| 29 |
model_paths = [
|
| 30 |
+
'karths/binary_classification_train_test',
|
|
|
|
| 31 |
"karths/binary_classification_train_process",
|
| 32 |
"karths/binary_classification_train_infrastructure",
|
| 33 |
"karths/binary_classification_train_documentation",
|
|
|
|
| 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',
|
| 46 |
'binary_classification_train_infrastructure': 'Infrastructure',
|
| 47 |
'binary_classification_train_documentation': 'Documentation',
|
|
|
|
| 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")
|