Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -27,23 +27,34 @@ login(token)
|
|
27 |
|
28 |
# Model paths and quality mapping
|
29 |
model_paths = [
|
30 |
-
'karths/
|
31 |
-
'karths/
|
32 |
-
"karths/
|
33 |
-
"karths/
|
34 |
-
"karths/
|
35 |
-
"karths/
|
36 |
-
"karths/
|
|
|
|
|
|
|
|
|
|
|
37 |
]
|
38 |
|
39 |
quality_mapping = {
|
40 |
-
'
|
41 |
-
'
|
42 |
-
'
|
43 |
-
'
|
44 |
-
'
|
45 |
-
'
|
46 |
-
'
|
|
|
|
|
|
|
|
|
|
|
|
|
47 |
}
|
48 |
|
49 |
# Pre-load models and tokenizer
|
|
|
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",
|
35 |
+
"karths/binary_classification_train_design",
|
36 |
+
"karths/binary_classification_train_defect",
|
37 |
+
"karths/binary_classification_train_code",
|
38 |
+
"karths/binary_classification_train_build",
|
39 |
+
"karths/binary_classification_train_automation",
|
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',
|
50 |
+
'binary_classification_train_design': 'Design',
|
51 |
+
'binary_classification_train_defect': 'Defect',
|
52 |
+
'binary_classification_train_code': 'Code',
|
53 |
+
'binary_classification_train_build': 'Build',
|
54 |
+
'binary_classification_train_automation': 'Automation',
|
55 |
+
'binary_classification_train_people': 'People',
|
56 |
+
'binary_classification_train_architecture':'Architecture'
|
57 |
+
|
58 |
}
|
59 |
|
60 |
# Pre-load models and tokenizer
|