happyHoliday commited on
Commit
9ccfe9f
1 Parent(s): 50ca014
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -31,7 +31,9 @@ dict_category_answer = {
31
 
32
 
33
  def QA(text):
34
-
 
 
35
  try:
36
  MODEL_NAME = 'cl-tohoku/bert-base-japanese-whole-word-masking'
37
  bert_sc = BertForSequenceClassification.from_pretrained("/saved_model")
 
31
 
32
 
33
  def QA(text):
34
+
35
+ return dict_category_answer['LABEL_0']
36
+
37
  try:
38
  MODEL_NAME = 'cl-tohoku/bert-base-japanese-whole-word-masking'
39
  bert_sc = BertForSequenceClassification.from_pretrained("/saved_model")