happyHoliday commited on
Commit
2b3b341
1 Parent(s): 9ccfe9f
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -32,7 +32,7 @@ dict_category_answer = {
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'
@@ -57,7 +57,7 @@ def QA(text):
57
  except e:
58
  return str(e)
59
 
60
- return "error occured"
61
 
62
  demo = gr.Interface(fn=QA, inputs="text", outputs="text")
63
 
 
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'
 
57
  except e:
58
  return str(e)
59
 
60
+ return answer_text
61
 
62
  demo = gr.Interface(fn=QA, inputs="text", outputs="text")
63