Update app.py

app.py

@@ -114,7 +114,7 @@ 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.models.squad import SquadLoss
+# from transformers.models.squad import SquadLoss
 
 
 # Load pre-trained model and tokenizer
@@ -170,9 +170,11 @@ train_data = prepare_training_data(qa_pairs)
 # Train the model
 learning_rate = 2e-5
 epochs = 3  # Adjust these values as needed
-model.compile(optimizer=AdamW(learning_rate=learning_rate), loss=SquadLoss())
+model.compile(optimizer=AdamW(learning_rate=learning_rate))
 model.fit(train_data, epochs=epochs)
 
+# loss=SquadLoss()
+
 # Save the trained model and tokenizer
 model.save_pretrained("bhagavad_gita_qa_model")
 tokenizer.save_pretrained("bhagavad_gita_qa_model")