Initial Commit

app.py

@@ -24,7 +24,7 @@ def inference(question,context):
     answer_start=torch.argmax(start_)
     answer_end=torch.argmax(end_)+1
     answer=tokenizer.convert_tokens_to_string(tokenizer.convert_ids_to_tokens(input_ids[answer_start:answer_end]))
-    print(answer)
+    return answer