karths commited on
Commit
92b2ff7
·
verified ·
1 Parent(s): f9c604d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +25 -14
app.py CHANGED
@@ -27,23 +27,34 @@ login(token)
27
 
28
  # Model paths and quality mapping
29
  model_paths = [
30
- 'karths/binary_classification_train_port',
31
- 'karths/binary_classification_train_perf',
32
- "karths/binary_classification_train_main",
33
- "karths/binary_classification_train_secu",
34
- "karths/binary_classification_train_reli",
35
- "karths/binary_classification_train_usab",
36
- "karths/binary_classification_train_comp"
 
 
 
 
 
37
  ]
38
 
39
  quality_mapping = {
40
- 'binary_classification_train_port': 'Portability',
41
- 'binary_classification_train_main': 'Maintainability',
42
- 'binary_classification_train_secu': 'Security',
43
- 'binary_classification_train_reli': 'Reliability',
44
- 'binary_classification_train_usab': 'Usability',
45
- 'binary_classification_train_perf': 'Performance',
46
- 'binary_classification_train_comp': 'Compatibility'
 
 
 
 
 
 
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