Update app.py

app.py

@@ -5,6 +5,7 @@ from torch.utils.data import DataLoader, RandomSampler, SequentialSampler
 from transformers import (AlbertConfig, AlbertForQuestionAnswering, AlbertTokenizer, squad_convert_examples_to_features)
 from transformers.data.processors.squad import SquadResult, SquadV2Processor, SquadExample
 from transformers.data.metrics.squad_metrics import compute_predictions_logits
+import gradio as gr
 
 model_name_or_path = "ktrapeznikov/albert-xlarge-v2-squad-v2"