Update app.py

app.py

@@ -113,7 +113,9 @@ for question in ["What is karma?", "Who is Arjuna?"]:
 # ]
 from transformers import BertTokenizer, TFBertForQuestionAnswering
 from transformers import AdamW  # Optimizer (optional)
-from transformers import SquadLoss  # Loss function (optional)
+# from transformers import SquadLoss  # Loss function (optional)
+from transformers.models.squad import SquadLoss
+
 
 # Load pre-trained model and tokenizer
 tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')