Change base model and remove title

app.py

@@ -9,7 +9,7 @@ from transformers import AutoModelForQuestionAnswering, AutoTokenizer
 MAX_SEQUENCE_LENGTH = 512
 
 models = {
-    "Base model": "bert-large-uncased-whole-word-masking-finetuned-squad",
+    "Base model": "madlag/bert-large-uncased-whole-word-masking-finetuned-squadv2",
     "Pruned model": "madlag/bert-large-uncased-wwm-squadv2-x2.63-f82.6-d16-hybrid-v1",
     "Pruned ONNX Optimized FP16": "tryolabs/bert-large-uncased-wwm-squadv2-optimized-f16",
 }
@@ -85,7 +85,6 @@ output_inference_time = gr.Text(label="Inference time in seconds")
 
 demo = gr.Interface(
     inference,
-    title="Optimizing Transformers - Question Answering Demo",
     inputs=[model_field, input_text_field, input_question_field],
     outputs=[output_model, output_inference_time],
 )