Update app.py

app.py

@@ -3,8 +3,7 @@ from transformers import DistilBertTokenizerFast, DistilBertForQuestionAnswering
 
 model_name = "distilbert-base-cased"
 tokenizer = DistilBertTokenizerFast.from_pretrained(model_name)
-# Set output_attentions and output_hidden_states to False to prevent weight initialization warnings
-model = DistilBertForQuestionAnswering.from_pretrained(model_name, output_attentions=False, output_hidden_states=False)
+model = DistilBertForQuestionAnswering.from_pretrained(model_name)
 
 def format_response(start_index, end_index, raw_answer):
     answer_tokens = tokenizer.convert_tokens_to_string([tokenizer.convert_ids_to_tokens(i)[0] for i in range(start_index, end_index+1)])