Update README.md

README.md

@@ -1,3 +1,13 @@
+---
+license: mit
+language:
+- en
+metrics:
+- bleu
+base_model:
+- google-bert/bert-base-uncased
+new_version: google-bert/bert-base-uncased
+---
 # Model Card for BERT-based Question Answering Model
 
 ## Model Details
@@ -48,4 +58,4 @@ def answer_question(question, context):
     start_idx = outputs.start_logits.argmax()
     end_idx = outputs.end_logits.argmax()
     answer = tokenizer.convert_tokens_to_string(tokenizer.convert_ids_to_tokens(inputs["input_ids"][0][start_idx:end_idx + 1]))
-    return answer
+    return answer