Update app.py

app.py

@@ -70,6 +70,7 @@ def main(question, context):
     # 编码输入
     text = question + " [SEP] " + context
     inputs = tokenizer(text, max_length=4096, truncation=True, return_tensors="pt")
+    print(question)
     # inputs.to('cuda')
     # 预测答案
     outputs = model(**inputs)
@@ -79,11 +80,13 @@ def main(question, context):
     # 解码答案
     is_impossible = null_scores.argmax().item()
     if is_impossible:
+        print('No Answer')
         return "No Answer"
     else:
         answer_start = torch.argmax(start_scores)
         answer_end = torch.argmax(end_scores) + 1
         answer = tokenizer.convert_tokens_to_string(tokenizer.convert_ids_to_tokens(inputs['input_ids'][0][answer_start:answer_end]))
+        print(answer)
         return answer