karths commited on
Commit
b8c4b59
·
verified ·
1 Parent(s): e2a073b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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")